LeetCode 825. Amigos apropiados para la edad (orden de recuento + suma de prefijo)

1. Título

Las personas se enviarán solicitudes de amistad entre sí, y ahora, dada una matriz que contiene su edad, edades [i] representa la edad de la i-ésima persona.

Cuando se cumple alguna de las siguientes condiciones, A no puede enviar una solicitud de amistad a B (A y B no son la misma persona):

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

De lo contrario, A puede enviar una solicitud de amistad a B.

Tenga en cuenta que si A envía una solicitud a B, no significa que B definitivamente enviará una solicitud a A.
Además, las personas no se enviarán solicitudes de amistad.

¿Cuántas solicitudes de amistad se enviarán en 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.

Fuente: LeetCode
Enlace: https://leetcode-cn.com/problems/friends-of-apropiado-ages Los
derechos de autor son propiedad de LeetCode . Para reimpresiones comerciales, comuníquese con la autorización oficial. Para reimpresiones no comerciales, indique la fuente.

2. Resolución de problemas

• Según las condiciones, las solicitudes solo se pueden enviar a partir de los 15 años.
• Cuente y ordene primero, y calcule el número de prefijos y
• Las personas de cada grupo de edad se dividen en dos partes: enviar a personas de su propia edad más joven y enviar a otros compañeros

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

---

Mi dirección de blog de CSDN https://michael.blog.csdn.net/

Mantenga presionado o escanee el código QR para seguir mi cuenta oficial (Michael Amin), ¡vengan juntos, aprendan y progresen juntos!