题目
难度:★☆☆☆☆
类型:数组
给定一副牌,每张牌上都写着一个整数。
此时,你需要选定一个数字 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)])
如有疑问或建议,欢迎评论区留言~