825. 适龄的朋友(Python)

2021-02-03  本文已影响0人  玖月晴

难度:★★☆☆☆
类型:数学
方法:无

力扣链接请移步本题传送门
更多力扣中等题的解决方案请移步力扣中等题目录

人们会互相发送好友请求,现在给定一个包含有他们年龄的数组,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.

提示:

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

解答

这个题目难度不大,对满足要求的情况进行统计即可。但是需要注意两点:

1.年龄既然是0到120范围内的整数,我们就不需要逐个人进行计算,而可以通过年龄段进行打包计算。

  1. 由于不能自己给自己发送好友,所以尤其需要注意重复统计的状况。
from collections import Counter


class Solution(object):
    def numFriendRequests(self, ages):
        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

如有疑问或建议,欢迎评论区留言~

有关更多力扣中等题的python解决方案,请移步力扣中等题解析

上一篇 下一篇

猜你喜欢

热点阅读