其他分享
首页 > 其他分享> > 从双倍数组中还原原数组

从双倍数组中还原原数组

作者:互联网

一个整数数组 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 不是一个双倍数组。

提示:

题解:注意考虑重复0的情况,还有其他重复数的情况。

code:

class Solution {
    public int[] findOriginalArray(int[] changed) {
        if (changed.length == 1) {
            return new int[0];
        }
        Map<Integer, Integer> map = new TreeMap<>();
        for (int num : changed) {
            if (map.containsKey(num)) {
                map.put(num, map.get(num) + 1);
            } else {
                map.put(num, 1);
            }
        }
        Set<Integer> skipSet = new HashSet<>();
        int k = 0;
        int[] result = new int[changed.length / 2];
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            Integer key = entry.getKey();
            Integer value = entry.getValue();
            if (key == 0) {
                 if (value % 2 == 0) {
                     for (int i = 0; i < value/2; i++) {
                        result[k++] = 0;
                     }
                     continue;
                 } else {
                     return new int[0];
                 } 
            }
            
            if (skipSet.contains(key)) {
                continue;
            }
            int sum = key * 2;
            if (map.containsKey(sum)) {
                Integer value2 = map.get(sum);
                if (value > value2) {
                    return new int[0];
                } else {
                    for (int i = 0; i < value; i++) {
                        result[k++] = key;
                    }
                    if (value == value2) {
                        skipSet.add(sum);
                    } else {
                        map.put(sum, value2 - value);
                    }
                    
                }
            } else {
                return new int[0];
            }
        }
        return result;
    }
}


 

标签:map,changed,int,双倍,value,还原,数组,new
来源: https://blog.csdn.net/xiao__jia__jia/article/details/120375467