题目描述
解法:
设 arr \textit{arr} arr 的长度为 n n n,题目本质上是问 arr \textit{arr} arr 能否分成 n 2 \dfrac{n}{2} 2n 对元素,每对元素中一个数是另一个数的两倍。设 cnt [ x ] \textit{cnt}[x] cnt[x] 表示 arr \textit{arr} arr 中 x x x 的个数
对于 arr \textit{arr} arr 中的 0 0 0,它只能与 0 0 0 匹配。如果 cnt [ 0 ] \textit{cnt}[0] cnt[0] 是奇数,那么必然无法满足题目要求
去掉 arr \textit{arr} arr 中的 0 0 0,设 x x x 为 arr \textit{arr} arr 中绝对值最小的元素,由于没有绝对值比 x x x 更小的数,因此 x x x 只能与 2 x 2x 2x 匹配。如果此时 cnt [ 2 x ] < cnt [ x ] \textit{cnt}[2x] < \textit{cnt}[x] cnt[2x]<cnt[x],那么会有部分 x x x 无法找到它的另一半,即无法满足题目要求;否则将所有 x x x 和 cnt [ x ] \textit{cnt}[x] cnt[x] 个 2 x 2x 2x 从 arr \textit{arr} arr 中去掉,继续判断剩余元素是否满足题目要求。不断重复此操作,如果某个时刻 arr \textit{arr} arr 为空,则说明 arr \textit{arr} arr 可以满足题目要求
代码实现时,我们可以用一个哈希表来统计 cnt \textit{cnt} cnt,并将其键值按绝对值从小到大排序,然后模拟上述操作,去掉元素的操作可以改为从 cnt \textit{cnt} cnt 中减去对应值
class Solution {
public:
bool canReorderDoubled(vector<int>& arr) {
unordered_map<int, int> cnt;
for (int x: arr) cnt[x]++;
if (cnt[0] % 2) return false;
vector<int> vals;
for (auto &[x, _]: cnt) vals.push_back(x);
sort(vals.begin(), vals.end(), [](int a, int b) { return abs(a) < abs(b); });
for (int x: vals)
{
if (cnt[2 * x] < cnt[x]) return false;
cnt[2 * x] -= cnt[x];
}
return true;
}
};