Problem: 914. 卡牌分组
刚开始没有读懂题目!后面才发现是要将相同牌面大小的数字分为一个组,然后在所有组的卡牌数量中(如分为1:2, 2:4,4又可以分为2,2),共同的不小于2的因子!!!
需要注意的是,刚刚想开始随便取两个数字把k取出来,但是发现180,390,855,而180和390取出来的因子有多个2/5/10,但是只有5是符合条件的,如果不把k的所有候选求出来,就不能通过所有测试集
Code
class Solution {
public:
vector<int> find_ks(int num1, int num2) {
vector<int> res;
for(int i = 1; i <= min(num1, num2); i ++) {
if(num1 % i == 0 && num2 % i == 0) {
res.push_back(i);
}
}
return res;
}
bool hasGroupsSizeX(vector<int>& deck) {
int n = deck.size();
map<int, int> nums;
for(int i = 0; i < n; i ++) {
nums[deck[i]] ++;
}
int l = nums.size();
// int cnt = 0;
// // find k
// int k = 10000;
// int k1 = 10000;
// for(map<int,int>::iterator it=nums.begin();it != nums.end();it++)
// {
// k = min(k, it->second);
// }
// // 本来是以为要找到最小的数字作为k。。。现在感觉是要找到所有数字除了1之外的最小因子
// for(map<int,int>::iterator it=nums.begin();it != nums.end();it++)
// {
// cnt ++;
// if(cnt == l) break;
// std::map<int,int>::iterator it2 = std::next(it);
// if((it->second != it2->second && (it->second % k != 0 ||it2->second % k != 0)) || it->second < 2 || it2->second < 2) return false;
// }
// if(l == 1 && nums[deck[0]] < 2) return false;
if(l == 1) {
return n >= 2;
}
else {
// find k
map<int,int>::iterator it1 = nums.begin();
map<int,int>::iterator it2 = std::next(it1);
int num1 = it1->second;
int num2 = it2->second;
vector<int> ks = find_ks(num1, num2);
int k = 1;
int ksize = ks.size();
for(int i = 0; i < ksize; i ++) {
// 验证其他数字是不是都可以使用这个数字整除
int flag = 0;
for(map<int,int>::iterator it=nums.begin();it != nums.end();it++)
{
if(it->second % ks[i] != 0) {
flag = 1;
continue;
}
}
if(!flag) k = ks[i];
}
if(k == 1) return false;
}
return true;
}
};