题目
在社交媒体网站上有 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 发送一条好友请求。另外,用户不会向自己发送好友请求。
返回在该社交媒体网站上产生的好友请求总数。
示例
-
示例一
输入:ages = [16,16]second = “good”
输出:2
解释:2 人互发好友请求。 -
示例二
输入:ages = [16,17,18]
输出:2
解释:产生的好友请求为 17 -> 16 ,18 -> 17 。 -
示例三
输入:ages = [20,30,100,110,120]
输出:3
解释:产生的好友请求为 110 -> 100 ,120 -> 110 ,120 -> 100 。
方法
双指针
class Solution {
public int numFriendRequests(int[] ages) {
Arrays.sort(ages);
int n = ages.length, ans = 0;
for (int k = 0, i = 0, j = 0; k < n; k++) {
while (i < k && !check(ages[i], ages[k])) i++;
if (j < k) j = k;
while (j < n && check(ages[j], ages[k])) j++;
if (j > i) ans += j - i - 1;
}
return ans;
}
boolean check(int x, int y) {
if (y <= 0.5 * x + 7) return false;
if (y > x) return false;
if (y > 100 && x < 100) return false;
return true;
}
}
- 对于当前用户y来说,符合条件的用户是一段连续的区间,且下界<y,上界>y,当y增大时,这个区间的上下界都是增大的。
- 因此,将ages数组先进行排序,然后使用l与r记录每一个用户的上下界,每遍历到一个新的用户,就对上下界更新,因为上下界总是在增大,所以只需要在上一个用户的上下界基础上进行更新就行。
- 每次更新完,r - l - 1就是发送的请求数。把其累加到ans上返回就是最终结果。