0
点赞
收藏
分享

微信扫一扫

825.适龄的朋友

沉浸在自己的世界里 2022-04-21 阅读 44

在社交媒体网站上有 n 个用户。给你一个整数数组 ages ,其中 ages[i] 是第 i 个用户的年龄。
如果下述任意一个条件为真,那么用户 x 将不会向用户 y(x != y)发送好友请求:

ages[y] <= 0.5 * ages[x] + 7
ages[y] > ages[x]
ages[y] > 100 && ages[x] < 100
否则,x 将会向 y 发送一条好友请求。

注意,如果 x 向 y 发送一条好友请求,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 。

提示:
n == ages.length
1 <= n <= 2 * 104
1 <= ages[i] <= 120

# 题目所给三个表达式最终可优化为 0.5 * ages[x] + 7 < ages[y] <= ages[x],若年龄<15则没有与其匹配的好友
class Solution:
    def numFriendRequests(self, ages: List[int]) -> int:
		# res代表最终结果,双指针,left代表可发送好友申请的起始位置,right代表终止位置,cur表示上次遍历时的年龄
        res = left = right = cur = curres =  0
        # 排序,便于计算,不影响最终结果
        ages.sort()
        for i in ages:
        	# 若当前遍历年龄和上次遍历年龄相同则不需要再去寻找,直接和上次结果相加即可
            if i == cur:
                res += curres 
                continue
            if i < 15: continue
            # 找到可发送好友申请的初始位置
            while ages[left] <= 0.5 * i + 7: left += 1
            # 找到可发送好友申请的终止位置
            while right + 1 < len(ages) and ages[right + 1] <= i: right += 1
            # 当前年龄可发送好友申请的次数
            curres = right - left
            # 将次数加至最终结果
            res += curres 
        
        return res
举报

相关推荐

0 条评论