Leetcode 825 适龄的朋友

题目

在社交媒体网站上有 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 发送一条好友请求。另外，用户不会向自己发送好友请求。

返回在该社交媒体网站上产生的好友请求总数。

解题思路

  哈希计数+前缀和。分析题目给的三个条件，发现第三个条件其实是包含在第二个条件中的，然后把前两个条件化简一下，就是 0.5 * age[x] + 7 < age[y] ≤ age[x] 这样一个区间，对于每个用户都要求这个区间内其他用户的数量。一开始准备排序+二分查找来写，看了数据范围只有 120 ，发现完全没必要排序，直接用哈希计数+前缀和就能搞定区间内用户的数量，时间复杂度只有O(n)。

代码

class Solution {
    public int numFriendRequests(int[] ages) {
        int[] pre = new int[130], plus = new int[130];
        int length = ages.length;
        for (int i = 0; i < length; i++) pre[ages[i]]++;
        for (int i = 1; i <= 120; i++) plus[i] += pre[i] + plus[i - 1];

        int ans = 0;
        for (int i = 15; i <= 120; i++) {
            int left = i / 2 + 7, right = i;
            ans += pre[i] * (plus[right] - plus[left] - 1);
        }
        return ans;
    }
}

  最后一个循环之所以从 15 开始，是因为要满足 0.5 * age[x] + 7 < age[y] ≤ age[x] ，算出来 age[x] > 14 。