力扣题825适龄的朋友

力扣题825适龄的朋友,第1张

力扣题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 发送一条好友请求。另外,用户不会向自己发送好友请求。

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

示例 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 * 104
  • 1 <= ages[i] <= 120

1.使用双重循环的暴力法,会超出时间限制,需要进行优化。

        相对于暴力解法对于每一个年龄都去判断,下面的优化其实就是对于每一个年龄找到一个可

      以交朋友的年龄范围,然后求和即可,可以减少很多不必要的判断,因此需要对题目条件进行

      进一步整合。

2.题解:

        排序+双指针:

class Solution {
    public int numFriendRequests(int[] ages) {
        if (ages == null || ages.length == 0 || ages.length == 1) {
            return 0;
        }
        //根据题目条件,对于一个人,他可以结交的朋友的年龄是在一个区间内的,因此排序后就可以找到每个人的区间
        Arrays.sort(ages);
        int left = 0;
        int right = 0;
        int count = 0;
        for (int age : ages) {
            if (age < 15) {
                continue;
            }
            while (ages[left] <= age * 0.5 + 7) { 
                left++;
            } 
            while (right + 1 < ages.length && ages[right + 1] <= age) {
                right++;
            }

            count += right - left;//因为区间内肯定包含自身,所以不用+1
        }
        return count;
    }
}

        计数排序+前缀和:

class Solution {
    public int numFriendRequests(int[] ages) {
        int[] countOfAges = new int[121];
        for (int age : ages) {//计算每个年龄出现的次数
            countOfAges[age]++;
        }

        int[] pre = new int[121];
        for (int i = 1;i <= 120;i++) {//计算每个年龄下的前缀和,即小于等于当前年龄的人数
            pre[i] = pre[i - 1] + countOfAges[i];
        }

        int count = 0;
        for (int i = 15;i <= 120;i++) {
            if (countOfAges[i] > 0) {
                int bound = (int)(i * 0.5 + 8);
                //对于每个年龄能申请朋友的数量乘以这个年龄的人的数量
                count += countOfAges[i] * (pre[i] - pre[bound - 1] - 1);//-1是为了减去自身
            }
        }

        return count;
    }
}

题源:力扣

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5682775.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存