914. 卡牌分组(leetcode)

题目描述

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/x-of-a-kind-in-a-deck-of-cards

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

此时，你需要选定一个数字 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（记得从2开始），判断是否有满足条件的 X 即可。
满足条件：X 是 ans（decx.size) 的约数 ， X 是每个数字i的数量count[i] 的约数

代码

class Solution {
public:
    bool hasGroupsSizeX(vector<int>& deck) {
        int ans = deck.size();
        int count[10010] = {0};
        for(int i = 0; i < deck.size(); i++) 
            count[deck[i]]++;
        vector<int> t;
        for(int i = 0; i < 10000; i++)
        {
            if(count[i] > 0)
                t.push_back(count[i]);
        }
        for(int x = 2; x <= ans; x++)
        {
            if(ans % x == 0)
            {
                bool flag = 1;
                for(int i = 0; i < t.size(); i++)
                {
                    if(t[i] % x != 0)
                    {
                        flag = 0;
                        break;
                    }
                }
                if(flag) return true;
            }
        }
        return false;
    }
};

上面说了X为count[i] 的约数，可以推出来，只有X是所有count[i]的最大公约数的约数时，才可能满足条件，因此我们可以求conut的最大公约数g，判断g是否大于2；

class Solution {
public:
    int gcd(int a, int b)
    {
        if(!a)
            return b;
        if(!b)
            return a;
        int c = a % b;
        while(c != 0)
        {
            a = b;
            b = c;
            c = a % b;
        }
        return b;
    }
    bool hasGroupsSizeX(vector<int>& deck) {
        int ans = deck.size();
        int count[10010] = {0};
        for(int i = 0; i < deck.size(); i++) 
            count[deck[i]]++;
       int g = -1;
       for(int i = 0; i < 10000; i++)
       {
           if(g == -1)
                g = count[i];
            else
                g = gcd(g,count[i]);
            
       }
       return g >= 2;
    }
};