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

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] 。
1
2
3
4
5
6
7

示例 2：

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

示例 3：

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

提示：

• 1 <= changed.length <= 105
• 0 <= changed[i] <= 105

解题思路与代码实现

    /**
     * 双倍数组升序排序
     * 使用哈希表记录双倍数组中所有元素出现频数
     * 如果一个元素它可以保存到结果中，会把它和它的两倍的频数减1，而频数为0的元素就不可以保存到结果中
     * 特别的测试用例: [0,0,0,0]
     * @param changed   双倍数组
     * @return  原数组
     */
    public int[] findOriginalArray(int[] changed) {
        int n = changed.length;
        if (n % 2 != 0) {
            // 空数组返回长度为0的数组
            return new int[0];
        }
        Arrays.sort(changed);   // 数组升序排序
        int[] res = new int[n / 2];   // 保存最终结果
        Map<Integer, Integer> map = new HashMap<>(); // 记录所有元素出现频数
        for (int i = 0; i < changed.length; i++) {
            map.put(changed[i], map.getOrDefault(changed[i], 0) + 1);
        }
        int j = 0;  // 辅助下标，用于给res数组赋值
        for (int i = 0; i < n ; i++) {
            // 频数为0的元素不会执行循环体
            if (map.get(changed[i]) == 0) {
                continue;
            }
            // 当前元素的频数减1，主要是特殊处理[0,0]这种重复元素的情况
            map.put(changed[i], map.get(changed[i]) - 1);
            // 当前元素的两倍
            int doubleTarget = changed[i] * 2;
            // 如果一个元素它的两倍的频数等于0，以[3,6]，[3,5]为例进行理解即可
            // [3,6]:当3被保存后，3,6的频数都变为了0，访问6的时候不会执行循环体
            // [3,5]:访问3执行循环体，找不到它的两倍，即changed数组不是双倍数组，返回空数组即可
            if (map.getOrDefault(doubleTarget, 0) == 0) {
                return new int[0];
            }
            // 如果一个元素它的两倍的频数大于0，可以将该元素它的两倍的频数-1，并把该元素保存到最终结果
            map.put(doubleTarget, map.get(doubleTarget) - 1);
            res[j++] = changed[i];
        }
        return res;
    }
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

踩坑点

空数组是返回new int[0]而非null