题目
在社交媒体网站上有 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 。
解题
注意题目中的不必,是指not necessary。因此没必要看这句话。
方法一:排序+滑动窗口
下面代码中x
就是age
,y就是(left,right)
范围内的人。
由于x
包含在(left,right)
中,因此最后还要减一,right-left-1
class Solution { public: int numFriendRequests(vector<int>& ages) { int n=ages.size(); sort(ages.begin(),ages.end()); int left=0,right=0,res=0; for(int age:ages){ if(age<15) continue; while(ages[left]<=0.5*age+7) left++; while(right<n&&ages[right]<=age) right++; res+=right-left-1;//不会向自己发送,因此y的范围(left,right)中包含着x(age),所以要减一 } return res; } };