开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第28天，点击查看活动详情
一 描述
在社交媒体网站上有 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 * 10^4
- 1 <= ages[i] <= 120
二 分析
- 先对数组进行排序
- 从索引为1处开始遍历
- 如果当前值 < 15 就直接continue（求解x = 0.5 * x + 7可知当值小于15时都不会满足）
- 先通过二分法找到大于0.5 * x + 7的值第一次出现的位置，然后结果直接加上两者索引之差
- 再通过二分法找到当前值第一次出现的位置（处理多个重复值的情况），同样加上两者索引之差
- 最后返回结果
注意点：
两个二分法所查找的值不一样，一个是找大于target的最小index，另一个是找target第一次出现的位置
三 答案
class Solution:
def numFriendRequests(self, ages: List[int]) -> int:
if not ages or len(ages) == 1:
return 0
ages.sort() # 排序
result = 0
for i in range(1, len(ages)):
if ages[i] < 15:
continue
left, right, target = 0, i, 0.5 * ages[i] + 7 # 找大于某个值的最小index
while left + 1 < right:
mid = (left + right) // 2
if ages[mid] > target:
right = mid
else:
left = mid
if ages[left] > target:
result += i - left
elif ages[right] > target:
result += i - right
left, right, target = 0, i, ages[i] # 找某个值第一次出现的index
while left + 1 < right:
mid = (left + right) // 2
if ages[mid] >= target:
right = mid
else:
left = mid
if ages[left] == target:
result += i - left
elif ages[right] == target:
result += i - right
return result
