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.
class Solution {
public int numFriendRequests(int[] ages) {
int[] nums = new int[121];
for (int age : ages) {
nums[age] ++;
}
int ans = 0;
for (int i = 120; i >= 1; i --) {
if (nums[i] == 0) {
continue;
}
if (i > 0.5 * i + 7) {
ans += nums[i] * (nums[i] - 1);
}
for (int j = i - 1; j >= 1; j --) {
if (j <= 0.5 * i + 7) {
break;
} else {
ans += nums[j] * nums[i];
}
}
}
return ans;
}
}