手记

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

解题思路


思路:求最大公约数。

审题,分析题意:

  • 相同整数的牌放在同一组
  • 每组的牌都是 X 张

这里首先想到的是统计元素,对照示例 1,示例 2,统计元素个数,元素个数相等可以分组成功返回 True,否则返回 False,这是一个思路。

再看示例 3,如果组内的元素只有 1 个的时候,直接返回 False。

再看示例 5,这里 1 的个数为 2 个, 2 的个数为 4 个。这里将原组 [2,2,2,2] 分成两组 [2,2],这样与前面 [1,1] 的元素个数就相同了。

针对示例 5,1 有 2 个,2 有 4 个,最终分组元素个数都是 2个,其实这里 X = 2,是这两个组元素个数的公约数。

根据上面结合题意与示例的分析,确定以下思路:

  • 首先遍历统计元素个数,如果某个元素只有一个,直接返回 False;
  • 考虑示例 5 的情况 ,拆分相同元素的组进而达到符合题意的分组。这里面涉及到公约数的思想,同时分组后组内的元素个数严格大于 1,那么可以考虑求最大公约数,同时确保最大公约数大于 1。

代码实现


from typing import List

class Solution:
    def hasGroupsSizeX(self, deck: List[int]) -> bool:
        # 用以统计元素个数
        from collections import Counter
        import math
        from functools import reduce
        
        # 计算元素的个数
        cnt_values = Counter(deck).values()
        # reduce() 对参数序列中的元素进行累积
        # 先对第 1,2 个元素进行操作,结果与第 3 个元素进行计算,以此类推
        return reduce(math.gcd, cnt_values) > 1

实现效果



以上就是通过统计元素个数,进而求最大公约数,并且在约束条件最大公约数严格大于 1 下,得出结果解决《卡牌分组》问题的主要内容。


1人推荐
随时随地看视频
慕课网APP