914. 卡牌分組(Python)

題目

難度:★☆☆☆☆
類型:數組

給定一副牌,每張牌上都寫著一個整數。

此時,你需要選定一個數字 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]

解答

卡牌可以分組的條件是:所有數字出現次數的最大公約數大于一。

根據這個原則,我們可以進行判斷:

  1. 首先統計每種卡牌數字出現的次數;

  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)])

如有疑問或建議,歡迎評論區留言~

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容