LeetCode 2007. 从双倍数组中还原原数组

题目描述

一个整数数组 original 可以转变成一个双倍数组 changed ，转变方式为将 original 中每个元素 值乘以 $2$ 加入数组中，然后将所有元素 随机打乱 。

给你一个数组 changed ，如果 change 是双倍数组，那么请你返回 original数组，否则请返回空数组。original 的元素可以以任意顺序返回。

示例 1：

输入：changed = [1,3,4,2,6,8]
输出：[1,3,4]
解释：一个可能的 original 数组为 [1,3,4] :
- 将 1 乘以 2 ，得到 1 * 2 = 2 。
- 将 3 乘以 2 ，得到 3 * 2 = 6 。
- 将 4 乘以 2 ，得到 4 * 2 = 8 。
其他可能的原数组方案为 [4,3,1] 或者 [3,1,4] 。

示例 2：

输入：changed = [6,3,0,1]
输出：[]
解释：changed 不是一个双倍数组。

示例 3：

输入：changed = [1]
输出：[]
解释：changed 不是一个双倍数组。

提示：

$1 \le changed.length \le 10^5$
$0 \le changed[i] \le 10^5$

排序 + Hash表

首先将数组从小到大排序，且记录每个元素出现的次数。

然后从小到大遍历每一个元素，对于一个元素 $x$ ，它出现的次数大于 $0$ 且 $x * 2$ 还在hash表中，就可可以将其加入答案数组中。

如果 $x$ 找不到 $x * 2$ ，即 $x * 2$ 出现的次数已经为 $0$ ，则表述无法找到原数组，返回空数组。

class Solution {
public:
    vector<int> findOriginalArray(vector<int>& changed) {
        sort(changed.begin(), changed.end());
        unordered_map<int, int> cnt;
        for ( auto x : changed ) cnt[x] ++;
        
        vector<int> res;
        for ( auto x : changed )
        {
            if ( cnt[x] == 0 ) continue;
            cnt[x] --;
            if ( cnt[x * 2] == 0 ) return {};
            cnt[x * 2] --;
            res.push_back(x);
        }
        return res;
    }
};