Leetcode 825. 适龄的朋友（双指针，单调性, 桶排序优化）

wwxy261

于 2021-12-27 17:21:58 发布

阅读量316

点赞数

分类专栏：算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/wwxy1995/article/details/122176225

版权

算法专栏收录该内容

3633 篇文章 120 订阅

订阅专栏

这篇博客讨论了如何优化解决好友申请问题的算法。原始方法复杂度为O(n^2)，但通过利用年龄的单调性，可以使用双指针或前缀和优化到线性时间复杂度。两种优化方案分别通过双指针遍历和计算前缀和来减少无效的请求计算，显著提高了效率。

摘要由CSDN通过智能技术生成

暴力做法是O(n^2), 检查所有人两两之间的关系。

优化方案利用年龄单调性，x只会像和他岁数一样或者比他小，到一定范围的人发送请求，因此可以用双指针优化。

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 (int age: ages) {
            if (age < 15) {
                continue;
            }
            while (ages[left] <= 0.5 * age + 7) {
                ++left;
            }
            while (right + 1 < n && ages[right + 1] <= age) {
                ++right;
            }
            ans += right - left;
        }
        return ans;
    }
};

class Solution {
public:
    int numFriendRequests(vector<int>& ages) {
        vector<int> cnt(121);
        for (auto age : ages) {
            ++cnt[age];
        }
        vector<int> pre(121);
        for (int i = 1; i <= 120; i++) {
            pre[i] = pre[i - 1] + cnt[i];
        }
        int res = 0;
        for (int i = 15; i <= 120; i++) {
                if (cnt[i]) {
                    int bound = i * 0.5 + 8;
                    res += cnt[i] * (pre[i] - pre[bound - 1] - 1);
                }
        }
        return res;
    }
};