leetcode 2007. 从双倍数组中还原原数组（C++、java、python）

一个整数数组 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 <= changed.length <= 105
• 0 <= changed[i] <= 105

C++

class Solution {
public:
    vector<int> findOriginalArray(vector<int>& changed) {
        int n=changed.size();
        unordered_map<int,int> mp;
        for(int i=0;i<n;i++) {
            mp[changed[i]]++;
        }
        sort(changed.begin(),changed.end());
        vector<int> res;
        for(int i=0;i<n;i++) {
            if(mp[changed[i]]>0) {
                mp[changed[i]]--;
                if(mp[changed[i]*2]>0) {
                    res.push_back(changed[i]);
                    mp[changed[i]*2]--;
                } else {
                    return vector<int>();
                }
            }
        }
        return res;
    }
};

java

class Solution {
    public int[] findOriginalArray(int[] changed) {
        ArrayList<Integer> res = new ArrayList<>();
        Map<Integer, Integer> mp = new HashMap<>();
        int n = changed.length;
        Arrays.sort(changed);
        for (int i = 0; i < n; i++) {
            mp.put(changed[i], mp.getOrDefault(changed[i], 0) + 1);
        }
        for (int i = 0; i < n; i++) {
            if (mp.get(changed[i]) > 0) {
                mp.put(changed[i], mp.get(changed[i]) - 1);
                if (mp.containsKey(changed[i] * 2) && mp.get(changed[i] * 2) > 0) {
                    mp.put(changed[i] * 2, mp.get(changed[i] * 2) - 1);
                    res.add(changed[i]);
                } else {
                    return new int[0];
                }
            }
        }
        int[] ans = new int[n / 2];
        for (int i = 0; i < ans.length; i++) {
            ans[i] = res.get(i);
        }
        return ans;
    }
}

python

扫描二维码关注公众号，回复： 13537977 查看本文章

class Solution:
    def findOriginalArray(self, changed: List[int]) -> List[int]:
        n = len(changed)
        ct = Counter()
        changed.sort()
        res = []
        for i in range(n):
            ct[changed[i]] += 1
        for i in range(n):
            if ct[changed[i]] > 0:
                ct[changed[i]] -= 1
                if ct[changed[i] * 2] > 0:
                    ct[changed[i] * 2] -= 1
                    res.append(changed[i])
                else:
                    return []
        return res