0
点赞
收藏
分享

微信扫一扫

leetcode-从双倍数组中还原原数组


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

java代码:

class Solution {
    public int[] findOriginalArray(int[] changed) {
        int n = changed.length;
        if (n % 2 == 1) return new int[]{};
        // 保证较小数先遍历到,因为最小的数肯定是原始数组里的
        Arrays.sort(changed);
        Map<Integer, Integer> cache = new HashMap<>();
        int cnt = 0;
        List<Integer> resList = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            if (cache.getOrDefault(changed[i], 0) > 0) {
                // 当前元素作为双倍元素匹配
                cnt++;
                cache.put(changed[i], cache.get(changed[i]) - 1);
                resList.add(changed[i] / 2);
            } else {
                // 当前元素作为原始元素匹配
                cache.put(changed[i] * 2, cache.getOrDefault(changed[i] * 2, 0) + 1);
            }
        }
        if (cnt != n / 2) return new int[]{};
        int[] res = new int[n / 2];
        for (int i = 0; i < res.length; i++) res[i] = resList.get(i);
        return res;
    }
}


举报

相关推荐

0 条评论