LeetCode #914 卡牌分组 数组

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 <= deck.length <= 10000
0 <= deck[i] < 10000

方法一

一开始想着就是用 HashMap 或者 大小为 10000 的数组统计值，这个思路是对的，比较时间复杂度只要 $O(1)$，求 X 的方法想错了：被例子带跑了，以为找到数量最少的那张牌能整除其他牌就对了，其实不是，因为数量最少的那张牌也是可以分成 N 份的，这就变成了求所有牌的数量的最大公约数的问题。

class Solution:
    def hasGroupsSizeX(self, deck: List[int]) -> bool:
    	# 求最大公约数
        def getGcd(x, y):
            return y if x % y == 0 else getGcd(y, x % y)
        counts = [0] * 10000
        for i in deck:
            counts[i]+=1
        # 随便取一个值出来
        gcd = counts[deck[0]]
        for i in counts:
            if i > 0:
            	# gcd存储当前最大公约数，一直和后面的数继续比下去
                gcd = getGcd(i, gcd)
                if gcd < 2:
                    return False
        return True 

时间复杂度：$O(NlogC)$，其中 $N$ 是卡牌的个数， $C$ 是数组 deck 中的数的范围，求两个数的最大公约数的复杂度是 $O(logC)$
空间复杂度：$O(N)$

方法二

python 用的还不是很熟悉，很多方法还不会用，官方的精简写法：

class Solution(object):
    def hasGroupsSizeX(self, deck):
        from fractions import gcd
        vals = collections.Counter(deck).values()
        return reduce(gcd, vals) >= 2