题目

类型:双指针
image.png

解题思路

从三个不发送好友请求的条件来看,以 y 的角度来说,可总结为:

  • 年龄比我小的不考虑(同龄的可以)
  • 年龄比我大可以考虑,但是不能超过一定范围则不考虑。

对于一个确定的 y 而言,会发送好友请求的 x 范围为连续段:

image.png

随着 y 的逐渐增大,对应的 x 连续段的左右边界均逐渐增大(数轴上均往右移动)。

因此,可以先对 ages 进行排序,枚举每个 y = ages[k] ,同时使用 i 和 j 维护左右区间,[i, j) 代表在 ages 上会往 y = ages[k] 发送请求的 x 连续段,统计每个 y = ages[k] 的 x 有多少个即是答案,同时需要注意在 [i, j) 范围内是包含 y = ages[k] 自身,统计区间长度时需要进行 -1 操作。

代码

  1. class Solution {
  2. int N = 130;
  3. public int numFriendRequests(int[] ages) {
  4. int[] nums = new int[N];
  5. for (int i : ages) nums[i]++;
  6. for (int i = 1; i < N; i++) nums[i] += nums[i - 1];
  7. int ans = 0;
  8. for (int y = 1, x = 1; y < N; y++) {
  9. int a = nums[y] - nums[y - 1]; // 有 a 个 y
  10. if (a == 0) continue;
  11. if (x < y) x = y;
  12. while (x < N && check(x, y)) x++;
  13. int b = nums[x - 1] - nums[y - 1] - 1; // [y, x) 为合法的 x 范围,对于每个 y 而言,有 b 个 x
  14. if (b > 0) ans += b * a;
  15. }
  16. return ans;
  17. }
  18. boolean check(int x, int y) {
  19. if (y <= 0.5 * x + 7) return false;
  20. if (y > x) return false;
  21. if (y > 100 && x < 100) return false;
  22. return true;
  23. }
  24. }