LeetCode 825.年齢に適した友達（カウントソート+プレフィックス合計）

1.タイトル

人々はお互いに友達リクエストを送信し、年齢を含む配列が与えられると、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.

出典：LeetCode
リンク：https：//leetcode-cn.com/problems/friends-of-appropriate-ages
著作権はLeetCodeが所有しています。商用の再版については、公式の承認に連絡してください。非商用の再版については、出典を示してください。

2.問題解決

• 条件により、15歳からの応募のみ可能です。
• 最初にカウントして並べ替え、プレフィックスの数を計算し、
• 各年齢層の人々は2つの部分に分けられます：彼ら自身のより若い年齢の人々に送ること、そして仲間のお互いに送ること。

class Solution {
    
    
public:
    int numFriendRequests(vector<int>& ages) {
    
    
        vector<int> age(121, 0);//每个年龄有多少人
        for(auto a : ages) 
            age[a]++;
        vector<int> sum(121,0);
        for(int i = 1; i <= 120; i++)
        {
    
    
            sum[i] += sum[i-1]+age[i];// 前缀人数和
        }
        int ans = 0, l, r;
        for(int i = 15; i <= 120; i++)
        {
    
    
            l = i/2 + 7;
            r = i;
            ans += (sum[r-1]-sum[l])*age[i]+age[i]*(age[i]-1);
            //  年龄小于当前的，满足条件的人* 每人都可以发
            //  + 当前同龄里的人，互相发，排列 An2
        }
        return ans;
    }
};

84 ms 28.6 MB

---

私のCSDNブログアドレスhttps://michael.blog.csdn.net/

QRコードを長押しまたはスキャンして、私の公式アカウント（Michael Amin）をフォローし、一緒に来て、学び、一緒に進歩してください！