题目:
给定一副牌,每张牌上都写着一个整数。
此时,你需要选定一个数字 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
思路:求出每个数出现的次数,要能满足分组,即所有的次数存在公约数而且这个公约数大于1.
代码:
第一版用 unordered_map 存每个数字出现的次数,但内存和时间都大许多
class Solution {
public:
//最大公约数
int gcd(int a, int b) {
int c = 0;
if (a < b) swap(a, b);
while (true) {
c = a % b;
if (c == 0) return b;
else
{
a = b;
b = c;
}
}
}
bool hasGroupsSizeX(vector<int>& deck) {
//存每个数的次数
unordered_map<int, int> um;
for (int a : deck) {
um[a] += 1;
}
//判断所有数有公约数时满足,公约数必须大于1
int maxY = (*(um.begin())).second;
for (unordered_map<int, int>::iterator it = um.begin(); it != um.end(); it++) {
if (it->second == 1) return false;
//后面的数要与前面的有公约数,所有数就可以整除,但要保证公约数大于1
maxY = gcd(maxY, it->second);
if (maxY < 2) return false;
}
return true;
}
};
第二版,由于提示中:0 <= deck[i] < 10000,所以用一维数组存次数,改善了一些
class Solution {
public:
//最大公约数
int gcd(int a, int b) {
int c = 0;
if (a < b) swap(a, b);
while (true) {
c = a % b;
if (c == 0) return b;
else
{
a = b;
b = c;
}
}
}
bool hasGroupsSizeX(vector<int>& deck) {
int num[10000] = {0};
for (auto x : deck) num[x]++;
int g =-1;
for (int i = 0; i < 10000; i++) {
if (num[i]) {
if (num[i] == 1) return false;
if (g !=-1) {
g = gcd(num[i], g);
}
else
{
g = num[i];
}
if (g == 1) return false;
}
}
return g >= 2 ? true : false;
}
};