leetcode954. 二倍数对数组（treemap）

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

示例 1：

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

代码

class Solution {
    public boolean canReorderDoubled(int[] A) {

        int n=A.length,res=0;
          if(n==0) return true;
        TreeMap<Integer,Integer> map=new TreeMap<>();
        for(int c:A) map.put(c,map.getOrDefault(c,0)+1);//初始化
        for(int c:map.keySet())
        {
            if(map.containsKey(2*c)&&map.get(c)>0&&map.get(c*2)>0) {
                
                int    t;
                if(c==0)//特殊情况
                t=map.get(0)/2;
                else t= Math.min(map.get(c),map.get(2*c));
                map.put(c,map.get(c)-t);//减掉已经匹配的数
                map.put(2*c,map.get(2*c)-t);
                res+=t;//记录匹配到的对数
                if(res>=n/2) return true;
            }
            
            
        }
        return false;
    }
}