跳转至

LeetCode: 825. 适龄的朋友

1、题目描述

人们会互相发送好友请求,现在给定一个包含有他们年龄的数组,ages[i] 表示第 i 个人的年龄。

当满足以下条件时,A 不能给 BA、B不为同一人)发送好友请求:

  • age[B] <= 0.5 * age[A] + 7
  • age[B] > age[A]
  • age[B] > 100 && age[A] < 100

否则,A 可以给 B 发送好友请求。

注意如果 AB 发出了请求,不等于 B 也一定会向 A 发出请求。而且,人们不会给自己发送好友请求。

求总共会发出多少份好友请求?

示例 1:

输入: [16,16]
输出: 2
解释: 二人可以互发好友申请。

示例 2:

输入: [16,17,18]
输出: 2
解释: 好友请求可产生于 17 -> 16, 18 -> 17.

示例 3:

输入: [20,30,100,110,120]
输出: 3
解释: 好友请求可产生于 110 -> 100, 120 -> 110, 120 -> 100.

说明:

  • 1 <= ages.length <= 20000.
  • 1 <= ages[i] <= 120.

2、解题思路

  • 统计出所有年龄的数量
  • 满足条件的直接计算即可

  • 需要注意同样年龄的处理,不能和自己发送朋友请求

from collections import Counter


class Solution:
    def numFriendRequests(self, ages: List[int]) -> int:
        ans = 0
        count = Counter(ages)

        for num, c in count.items():
            for num1, c1 in count.items():
                if num1 <= (num * 0.5 + 7):
                    continue
                if num1 > num:
                    continue
                if num < 100 < num1:
                    continue
                ans += c * c1
                if num == num1:
                    ans -= c
        return ans