【LeetCode】954. 二倍数对数组

题目

954. 二倍数对数组

给定一个长度为偶数的整数数组 arr，只有对 arr 进行重组后可以满足 “对于每个 0 <= i < len(arr) / 2，都有 arr[2 * i + 1] = 2 * arr[2 * i]” 时，返回 true；否则，返回 false。

示例 1：

输入：arr = [3,1,3,6]
输出：false

示例 2：

输入：arr = [2,1,2,6]
输出：false

示例 3：

输入：arr = [4,-2,2,-4]
输出：true
解释：可以用 [-2,-4] 和 [2,4] 这两组组成 [-2,-4,2,4] 或是 [2,4,-2,-4]

提示：

• 0 <= arr.length <= 3 * 104
• arr.length 是偶数
• -105 <= arr[i] <= 105

思路

• 将数组升序排列，左侧的数只可能匹配其右侧的数，所以升序排列后可以从左到右遍历数组
• 因为0可以匹配它自身，所以单独考虑，匹配数为$\lfloor numberofzero÷2\rfloor$

代码

from collections import Counter
class Solution:
    def canReorderDoubled(self, arr: List[int]) -> bool:
        arrCounter = Counter(arr)
        count = 0
        arr.sort()
        # zero can be paired to itself
        count += arrCounter[0] // 2
        arrCounter[0] = 0
        for n in arr:
            if count == len(arr)//2: return True
            if arrCounter[2*n] > 0 and arrCounter[n] > 0:
                arrCounter[2*n] -= 1
                arrCounter[n] -= 1
                count += 1
        return count == len(arr)//2

复杂度

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$