LeetCode 954. 二倍数对数组

题目描述

954. 二倍数对数组

解法：

设 $\text{arr}$ 的长度为 $n$，题目本质上是问 $\text{arr}$ 能否分成 $\frac{n}{2}$ 对元素，每对元素中一个数是另一个数的两倍。设 $\text{cnt}[x]$ 表示 $\text{arr}$ 中 $x$ 的个数

对于 $\text{arr}$ 中的 $0$，它只能与 $0$ 匹配。如果 $\text{cnt}[0]$ 是奇数，那么必然无法满足题目要求

去掉 $\text{arr}$ 中的 $0$，设 $x$ 为 $\text{arr}$ 中绝对值最小的元素，由于没有绝对值比 $x$ 更小的数，因此 $x$ 只能与 $2x$ 匹配。如果此时 $\text{cnt}[2x]<\text{cnt}[x]$，那么会有部分 $x$ 无法找到它的另一半，即无法满足题目要求；否则将所有 $x$ 和 $\text{cnt}[x]$ 个 $2x$ 从 $\text{arr}$ 中去掉，继续判断剩余元素是否满足题目要求。不断重复此操作，如果某个时刻 $\text{arr}$ 为空，则说明 $\text{arr}$ 可以满足题目要求

代码实现时，我们可以用一个哈希表来统计 $\text{cnt}$，并将其键值按绝对值从小到大排序，然后模拟上述操作，去掉元素的操作可以改为从 $\text{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;
    }
};