題目
難度:★☆☆☆☆
類型:數組
給定一副牌,每張牌上都寫著一個整數。
此時,你需要選定一個數字 X,使我們可以將整副牌按下述規則分成 1 組或更多組:
每組都有 X 張牌。
組內所有的牌上都寫著相同的整數。
僅當你可選的 X >= 2 時返回 true。
提示
1 <= deck.length <= 10000
0 <= deck[i] < 10000
示例
示例 1
輸入:[1,2,3,4,4,3,2,1]
輸出:true
解釋:可行的分組是 [1,1],[2,2],[3,3],[4,4]
示例 2
輸入:[1,1,1,2,2,2,3,3]
輸出:false
解釋:沒有滿足要求的分組。
示例 3
輸入:[1]
輸出:false
解釋:沒有滿足要求的分組。
示例 4
輸入:[1,1]
輸出:true
解釋:可行的分組是 [1,1]
示例 5
輸入:[1,1,2,2,2,2]
輸出:true
解釋:可行的分組是 [1,1],[2,2],[2,2]
解答
卡牌可以分組的條件是:所有數字出現次數的最大公約數大于一。
根據這個原則,我們可以進行判斷:
首先統計每種卡牌數字出現的次數;
分別驗證各種數字出現的次數是否都能被2、3、4...10001整除,這里的10000來自于題目對輸入長度的限制;如果遍歷過程中發現任意一個數字n(2<=n<=10000),那么卡牌就可以每n個分成一組,如果沒有找到這樣的數字,說明無法分組。
我們使用Counter進行計數。
class Solution:
def hasGroupsSizeX(self, deck):
"""
:param deck: List[int]
:return: bool
"""
from collections import Counter
deck = Counter(deck)
n = len(deck)
for i in range(2, 10001):
count = 0
for c in deck.values():
if c % i == 0: # i是f的最大公約數
count += 1
else:
break
if count == n:
return True
return False
下面是緊湊寫法,需要列表推導式、all、any內置函數等python基礎知識。
class Solution:
def hasGroupsSizeX(self, deck):
"""
:param deck: List[int]
:return: bool
"""
from collections import Counter
deck = Counter(deck)
return any([all([c % i == 0 for c in deck.values()]) for i in range(2, 10001)])
如有疑問或建議,歡迎評論區留言~