利用哈希表+排序解决“二倍数对数组”问题

Posted 2023-02-06 &小小白&

二十六、二倍数对数组

26.1、题设要求

  给定一个长度为偶数的整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/array-of-doubled-pairs

26.2、解题思路

  先将数组中的数放到map中,再将数组进行排序.
  如果数组中的数小于零,右移一位并判断map中是否有对应的数;如果数组中的数大于零，左移一位并判断map中是否有对应的数.

26.3、算法

class Solution 
    public boolean canReorderDoubled(int [] arr)
        final Map<Integer,Integer> map = new HashMap<>();
        //将数组中的每个数加到map中
        for (int a:arr) 
            map.put(a, 1 + map.getOrDefault(a, 0));
        
        //将数组排序
        Arrays.sort(arr);
        for (int a:arr) 
            if (map.get(a) == 0)
                continue;
            
            //如果a小于零,a右移一位;如果a大于零,a左移一位
            final int aa = a < 0 ? a >> 1 : a << 1;
            //a为负数的情况下,如果a对2取余不等于0或者aa除以2之后没有对应的数,直接返回false
            if (a < 0)
                //a对2取余不等于0
                if (a % 2 != 0 || map.getOrDefault(aa, 0) <= 0)
                    return false;
                
            else //a为正数的情况下,如果aa乘2之后没有对应的数,直接返回false
                if (map.getOrDefault(aa, 0) <= 0)
                    return false;
                
            
            //将两个数在map中计数都减一
            map.put(a, -1 + map.get(a));
            map.put(aa, -1 + map.get(aa));
        
        return true;
    

参考视频：B站up主代码魔术师