leetcode825. 适龄的朋友

leetcode825. 适龄的朋友

---

人们会互相发送好友请求，现在给定一个包含有他们年龄的数组，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.

思路

类似于桶排序，同一个年龄的人放在同一个桶里面，这样2W人就可以变成容量为120的数组。

public int numFriendRequests(int[] ages) {
        int count=0;
        int[] arr=new int[121];
        for(int i=0;i<ages.length;i++) {
        	arr[ages[i]]++;
        }
        for(int i=1;i<arr.length;i++) {
        	if(arr[i]==0)continue;
            if(i>0.5*i+7)count+=(arr[i]-1)* arr[i];
        	for(int j=1;j<arr.length&&j!=i;j++) {
        		if(arr[j]==0)continue;
        		if(j<=0.5*i+7)continue;
        		//if(j<=(i>>>1)+7)用了移位操作直接7ms->4ms
        		if(j>i)break;
        		if(j>100&&i<100)break;
        		count+=arr[i]*arr[j];
        	}
        }
        return count;
    }

leetcode 19/100