leetcode_825 適齡的朋友

語言: CN / TW / HK

要求

人們會互相傳送好友請求,現在給定一個包含有他們年齡的陣列,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.

核心程式碼

python class Solution: def numFriendRequests(self, ages: List[int]) -> int: from collections import Counter ages = Counter(ages) ans = 0 for ageA,countA in ages.items(): for ageB,countB in ages.items(): if ageA * 0.5 + 7 >= ageB or ageA < ageB or ageA < 100 < ageB: continue ans += countA * countB if ageA == ageB: ans -= countA return ans

image.png

解題思路:此題比較簡單,我們先對相同年齡進行統計,我們做兩輪迴圈,我們將不符合上面條件的都進行剔除,然後我們以相同年齡為一組,組和組之間可以交流,所有的可能的方式有,假設3歲5人,4歲3人,5人和3人都會打招呼,一共可打招呼5 * 3,當都是3歲的兩組人(5 - 1) * 5,除去自己可以給4個人打招呼,我們需要將自己減掉,所以和上面流程中的程式碼一樣。