题目描述
给定一副牌,每张牌上都写着一个整数。
此时,你需要选定一个数字 X,使我们可以将整副牌按下述规则分成 1 组或更多组:
- 每组都有 X 张牌。
- 组内所有的牌上都写着相同的整数。
仅当你可选的 X >= 2 时返回 true。
示例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]
提示:
1 <= deck.length <= 10000
0 <= deck[i] < 10000
解题思路
- 首先统计下每个数字的出现次数(我们只需关心出现的次数,不关心数字本身是多少)
- 然后求解这些分组的最大公约数
- 如果大于1,说明我们可以找到这样的分组
代码实现
class Solution:
def hasGroupsSizeX(self, deck: List[int]) -> bool:
def gcd(a, b):
if b == 0: return a
return gcd(b, a % b)
return reduce(gcd, collections.Counter(deck).values()) > 1
复杂度分析
时间复杂度:O(N)O(N)
空间复杂度:空间复杂度取决于递归的深度,因此空间复杂度为 O(log(max(a, b)))O(log(max(a,b)))。如果你将上述过程改成迭代,那么可以降低到O(1)O(1),也不难,大家可以自己试试。