[LeetCode]914. 卡牌分组

题目

给定一副牌，每张牌上都写着一个整数。

此时，你需要选定一个数字 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. 1 <= deck.length <= 10000
2. 0 <= deck[i] < 10000

解题思路

一开始我是先取得不同种类中出现次数的最小个数，判断的时候用所有出现的次数来取余这个最小值，后来一个测试样例过不了，才发现原来是求不同种类之间出现个数的最大公约数。所以只需计算出不同种类次数间的最大公约数X即可。如果X大于等于2则返回True，否则返回False。
复杂度分析：
时间复杂度：O(NlogC)，其中 N 是卡牌的个数，C 是数组 deck 中数的范围，在本题中 C 的值为 10000。求两个数最大公约数的复杂度是 O(logC)，需要求最多 N-1 次。
空间复杂度：O(N + C) 或 O(N)。

代码

class Solution:
    def gcd(self, a, b):
        return a if b==0  else gcd(b, a%b)

    def hasGroupsSizeX(self, deck: List[int]) -> bool:
        if not deck:
            return Fasle
        dic = {}
        for i in deck:
            if i in dic:
                dic[i] += 1
            else:
                dic[i] = 1
        nums = []
        for k in dic:
            nums.append(dic[k])
        x  = nums[0]
        for j in range(1, len(nums)):
            x = gcd(x, nums[j])
        return True if x>=2 else False