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;
}
};