914. 卡牌分组 C++ 5行 改进官方题解 极度简洁

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

思路

昨晚过0点,想做道题睡觉。最开始写的根据数字计数,所有数量相同则为真。提交出错,看到[1,1,2,2,2,2]就明白了,考虑还是不周到。想了一下,只要所有的牌数存在>=2的公约数即可。提交后时间内存都是最后的10%,非常尴尬。以下仅作记录,建议去看官方题解。
这篇文章写一半的时候我测试了官方代码,也是后10%,我收回上面的话,哈哈哈。
官方题解比我来得简洁很多,不过我觉得官方题解仍有改进空间。
任意数与0的gcd是自身,利用这个特性可以直接跳过了所有数量为0的牌,省去cnt[i]非零的判断,同时循环内的if(~g)也可以省去,代码极度简洁,方法非常巧妙。
任意数与0的gcd是自身,这句话可能让你感到疑惑,不要着急,这里需要讲一下gcd的实现。
在我不知道C++自带gcd的时候我自己的gcd长这样,看完你就明白了。

inline int gcd(int a, int b) {
    if (b) while ((a %= b) && (b %= a));
    return a + b;
}

C++内建的__gcd()估计也是碾转相除法。
还有官方题解的这个评论让我笑出了声。
在这里插入图片描述

改进的官方题解

class Solution {
    int cnt[10000];
public:
    bool hasGroupsSizeX(vector<int>& deck) {
        for (auto x: deck) cnt[x]++;
        int g = 0;
        for (int i = 0; i < 10000; ++i)
            g = gcd(g, cnt[i]);
        return g >= 2;
    }
};

我的(后10% )解答

class Solution {
public:
    bool hasGroupsSizeX(vector<int>& deck) {
        map<int, int> cards;
        for (auto num:deck){
            if (cards.count(num) != 0)
                cards[num]++;
            else
                cards[num] = 1;
        }
        int cnt = cards.begin()->second;
        for (auto card : cards)
            if (card.second != cnt)
                return false;
        return true;
    }
};
发布了38 篇原创文章 · 获赞 0 · 访问量 1016

猜你喜欢

转载自blog.csdn.net/Cyan1956/article/details/105134780