题目链接: 适龄的朋友
有关题目
在社交媒体网站上有 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 * 10^4
1 <= ages[i] <= 120
题解
法一:排序 + 双指针
参考官方题解
Tips
class Solution {
public:
int numFriendRequests(vector<int>& ages) {
int n = ages.size();
sort(ages.begin(), ages.end());
//双指针
int left = 0, right = 0, ans = 0;
for (auto &age : ages)
{
if (age < 15)
continue;
while(left < n && ages[left] <= 0.5 * age + 7)
left++;
//跳出循环说明ages[right] <= age
//ages[right + 1] > age
while(right + 1 < n && ages[right + 1] <= age)
right++;
ans += right - left;
}
return ans;
}
};