Java實現 LeetCode 825 適齡的朋友(暴力)

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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章