列表

详情


825. 适龄的朋友

在社交媒体网站上有 n 个用户。给你一个整数数组 ages ,其中 ages[i] 是第 i 个用户的年龄。

如果下述任意一个条件为真,那么用户 x 将不会向用户 yx != y)发送好友请求:

否则,x 将会向 y 发送一条好友请求。

注意,如果 xy 发送一条好友请求,y 不必也向 x 发送一条好友请求。另外,用户不会向自己发送好友请求。

返回在该社交媒体网站上产生的好友请求总数。

 

示例 1:

输入:ages = [16,16]
输出:2
解释:2 人互发好友请求。

示例 2:

输入:ages = [16,17,18]
输出:2
解释:产生的好友请求为 17 -> 16 ,18 -> 17 。

示例 3:

输入:ages = [20,30,100,110,120]
输出:3
解释:产生的好友请求为 110 -> 100 ,120 -> 110 ,120 -> 100 。

 

提示:

原站题解

去查看

上次编辑到这里,代码来自缓存 点击恢复默认模板
class Solution { public: int numFriendRequests(vector<int>& ages) { } };

python3 解法, 执行用时: 104 ms, 内存消耗: 14.1 MB, 提交时间: 2020-12-03 23:48:13

class Solution:
    def numFriendRequests(self, ages: List[int]) -> int:
        map_ = dict()
        for age in ages:
            if age in map_.keys():
                map_[age] += 1
            else:
                map_[age] = 1
        ans = 0
        for ageA, countA in map_.items():
            for ageB, countB in map_.items():
                if ageB <= 0.5 * ageA + 7: continue
                if ageB > ageA: continue
                if ageB > 100 and ageA < 100: continue
                ans += countA * countB
                if ageA == ageB:
                    ans -= countA
        return ans

上一题