要求
人们会互相发送好友请求,现在给定一个包含有他们年龄的数组,ages[i] 表示第 i 个人的年龄。
当满足以下任一条件时,A 不能给 B(A、B不为同一人)发送好友请求:
- age[B] <= 0.5 * age[A] + 7
- age[B] > age[A]
- age[B] > 100 && age[A] < 100 否则,A 可以给 B 发送好友请求。
注意如果 A 向 B 发出了请求,不等于 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.
核心代码
class Solution:
def numFriendRequests(self, ages: List[int]) -> int:
from collections import Counter
ages = Counter(ages)
ans = 0
for ageA,countA in ages.items():
for ageB,countB in ages.items():
if ageA * 0.5 + 7 >= ageB or ageA < ageB or ageA < 100 < ageB:
continue
ans += countA * countB
if ageA == ageB:
ans -= countA
return ans
解题思路:此题比较简单,我们先对相同年龄进行统计,我们做两轮循环,我们将不符合上面条件的都进行剔除,然后我们以相同年龄为一组,组和组之间可以交流,所有的可能的方式有,假设3岁5人,4岁3人,5人和3人都会打招呼,一共可打招呼5 * 3,当都是3岁的两组人(5 - 1) * 5,除去自己可以给4个人打招呼,我们需要将自己减掉,所以和上面流程中的代码一样。