825. 适龄的朋友
在社交媒体网站上有 n 个用户。给你一个整数数组 ages ,其中 ages[i] 是第 i 个用户的年龄。
如果下述任意一个条件为真,那么用户 x 将不会向用户 y(x != y)发送好友请求:
age[y] <= 0.5 * age[x] + 7
age[y] > age[x]
age[y] > 100 && age[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解题思路
- 排序+双指针。
- 观察题目给出的三个条件。可以推测出,用户y需要满足0.5×ages[x]+7,x用户才一定可以像y用户发送好友请求。
- 当age[x]<=14时,不存在满足要求的用户y,所以我们需要满足age[x]>=15的情况,此时的age[y]的范围是(0.5×ages[x]+7,ages[x]]。
- 然后通过维护左右边界进行遍历,找出答案。
class Solution:
def numFriendRequests(self, ages: List[int]) -> int:
n = len(ages)
ages.sort()
left = right = res = 0
for age in ages:
if age < 15:
continue
while ages[left] <= 0.5 * age + 7:
left += 1
while right + 1 < n and ages[right + 1] <= age:
right += 1
res += right - left
return res
运行结果



