每日算法系列【LeetCode 825】适龄的朋友

题目描述

人们会互相发送好友请求，现在给定一个包含有他们年龄的数组，ages[i] 表示第 i 个人的年龄。

当满足以下条件时，A 不能给 B（A、B不为同一人）发送好友请求：

• age[B] <= 0.5 * age[A] + 7
• age[B] > age[A]
• age[B] > 100 && age[A] < 100

否则，A 可以给 B 发送好友请求。

注意如果 A 向 B 发出了请求，不等于 B 也一定会向 A 发出请求。而且，人们不会给自己发送好友请求。

求总共会发出多少份好友请求?

示例1

输入：
[16,16]
输出：
2
解释：
二人可以互发好友申请。

示例2

输入：
[16,17,18]
输出：
2
解释：
好友请求可产生于 17 -> 16, 18 -> 17.

示例3

输入：
[20,30,100,110,120]
输出：
3
解释：
好友请求可产生于 110 -> 100, 120 -> 110, 120 -> 100.

提示

• 1 <= ages.length <= 20000.
• 1 <= ages[i] <= 120.

题解

计数法

代码

c++

class Solution {
public:
    int numFriendRequests(vector<int>& ages) {
        const int MA = 120;
        vector<int> count(MA+1, 0), sum(MA+1, 0);
        int n = ages.size(), res = 0;
        for (int i = 0; i < n; ++i) {
            count[ages[i]]++;
        }
        for (int i = 1; i <= MA; ++i) {
            sum[i] = sum[i-1] + count[i];
        }
        for (int b = 15; b <= MA; ++b) {
            res += count[b] * (sum[min(MA, 2*b-15)] - sum[b-1] - 1);
        }
        return res;
    }
};

python

class Solution:
    def numFriendRequests(self, ages: List[int]) -> int:
        MA = 120
        count, S = [0]*(MA+1), [0]*(MA+1)
        n, res = len(ages), 0
        for age in ages:
            count[age] += 1
        for i in range(1, MA+1):
            S[i] = S[i-1] + count[i]
        for b in range(15, MA+1):
            res += count[b] * (S[min(MA, 2*b-15)] - S[b-1] - 1)
        return res