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

解法

我们先将 changed 排序，然后从左往右遍历。合格的话应该满足如下情况。

对于一个元素来说：

如果它是 original 元素，那么应该至少有一个 double 后的元素在后面，我们将其标记。

如果它是 double 后的元素，那么它应该在我们之前访问到其 original 值时已经被标记过了

如果出现不在二者之列的元素，则该 changed 数组不合格，返回空。

代码


class Solution:
    def findOriginalArray(self, changed: List[int]) -> List[int]:
        changed.sort()
        n = len(changed)
        if n % 2 != 0:
            return []
        ans = []
        hash = collections.defaultdict(list)
        for i in range(n):
            hash[changed[i]].append(i)
        used = [False] * n
        for i in range(n):
            if used[i]:
                continue
            num = changed[i]
            twice = num * 2
            used[i] = True
            paired = False
            while hash[twice]:
                j = hash[twice].pop()
                if not used[j]:
                    used[j] = True
                    paired = True
                    break
            if paired:
                ans.append(num)
            else:
                return []
        return ans