954. 二倍数对数组
难度中等
给定一个长度为偶数的整数数组 A,只有对 A 进行重组后可以满足 “对于每个 0 <= i < len(A) / 2,都有 A[2 * i + 1] = 2 * A[2 * i]” 时,返回 true;否则,返回 false。
示例 1:
输入:[3,1,3,6]
输出:false
示例 2:
输入:[2,1,2,6]
输出:false
示例 3:
输入:[4,-2,2,-4]
输出:true
解释:我们可以用 [-2,-4] 和 [2,4] 这两组组成 [-2,-4,2,4] 或是 [2,4,-2,-4]
示例 4:
输入:[1,2,4,16,8,4]
输出:false
提示:
1. 0 <= A.length <= 30000
2. A.length 为偶数
3. -100000 <= A[i] <= 100000
思路1.0:
设置哈希表<int val,int num> pair,其中val为数组中的元素,num为该元素出现的次数。
对元素序列进行sort()排序,每次有新的元素加入时,通过pair.find(val/2) 与 pair.find(val*2)来确认是否存在有和他配对的值,如果存在,则将两数对的num–,在最后通过检查所有的pair中的num是否全部归零来判断答案
代码1.0(超时):
class Solution {
public:
bool canReorderDoubled(vector<int>& A) {
sort(A.begin(), A.end());
map<int, int> pair;
//将A中的信息转移到pair中
for (const auto& i : A) {
pair[i]++;
}
for (auto& i : pair) {
if (i.second < 0) return false;
else if (i.second == 0) continue;
map<int, int>::iterator mate;
if (i.first > 0) mate = pair.find(i.first * 2);
else {
if (i.first % 2 != 0) return false;
mate = pair.find(i.first / 2);
}
//如果没找到该元素的配对项
if (mate == pair.end()) return false;
//如果找到该元素的配对项
else {
while (i.second != 0) {
i.second--;
mate->second--;
}
}
}
return true;
}
};
在A的数量非常庞大的时候会超时,(A中存在大量重复元素)。
经排查,发现是多此一举的sort()函数与 找到元素配对项时进行的憨憨处理(每次-1).
代码2.0:
class Solution {
public:
bool canReorderDoubled(vector<int>& A) {
map<int, int> pair;
//将A中的信息转移到pair中
for (const auto& i : A) {
pair[i]++;
}
for (auto& i : pair) {
if (i.second < 0) return false;
else if (i.second == 0) continue;
map<int, int>::iterator mate;
if (i.first > 0) mate = pair.find(i.first * 2);
else {
if (i.first % 2 != 0) return false;
mate = pair.find(i.first / 2);
}
//如果没找到该元素的配对项
if (mate == pair.end()) return false;
//如果找到该元素的配对项
else {
int max = i.second;
mate->second -= max;
}
}
return true;
}
};