LeetCode 825. Amis adaptés à l'âge (tri par nombre + somme des préfixes)

1. Titre

Les gens vont s'envoyer des demandes d'amis, et maintenant, étant donné un tableau contenant leur âge, ages [i] représente l'âge de la i-ème personne.

Lorsque l'une des conditions suivantes est remplie, A ne peut pas envoyer de demande d'amitié à B (A et B ne sont pas la même personne):

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

Sinon, A peut envoyer une demande d'ami à B.

Notez que si A envoie une demande à B, cela ne signifie pas que B enverra définitivement une demande à A.
De plus, les gens ne s'enverront pas de demandes d'amis.

Combien de demandes d'amis seront envoyées au total?

示例 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.

Source: LeetCode
Lien: https://leetcode-cn.com/problems/friends-of-appropriate-ages Le
droit d'auteur est la propriété de LeetCode . Pour les réimpressions commerciales, veuillez contacter l'autorisation officielle. Pour les réimpressions non commerciales, veuillez indiquer la source.

2. Résolution de problèmes

• Selon les conditions, les candidatures ne peuvent être envoyées qu'à partir de 15 ans
• Comptez et triez d'abord, puis calculez le nombre de préfixes et
• Les personnes de chaque groupe d'âge sont divisées en deux parties: envoyer à des personnes de leur plus jeune âge et s'envoyer entre pairs

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 Mo

---

Mon adresse de blog CSDN https://michael.blog.csdn.net/

Appuyez longuement ou scannez le code QR pour suivre mon compte officiel (Michael Amin), venez ensemble, apprenez et progressez ensemble!