Leetcode 954. 二倍数对数组

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

示例 1：

输入：[3,1,3,6]
输出：false
示例 2：

输入：[2,1,2,6]
输出：false
示例 3：

输入：[4,-2,2,-4]
输出：true
解释：我们可以用 [-2,-4] 和 [2,4] 这两组组成 [-2,-4,2,4] 或是 [2,4,-2,-4]
示例 4：

输入：[1,2,4,16,8,4]
输出：false


提示：

0 <= A.length <= 30000
A.length 为偶数
-100000 <= A[i] <= 100000

需要注意的是负数时，对应的倍数是二分之一，而正数是一半。

class Solution {
public:
    bool canReorderDoubled(vector<int>& A) {
        map<int,int> num_map;
        for (auto &it : A){
            if(num_map.count(it)==0){
                num_map[it]=1;
            }else{
                num_map[it] +=1;
            }
        }
        bool ans = true;
        for (auto &num_pair:num_map){
            while(num_map[num_pair.first]>0){
                int match_num;
                if(num_pair.first>0) match_num = num_pair.first*2;
                else match_num = num_pair.first/2;
                if(num_map.count(match_num)&&(num_map[match_num]>0)){
                    num_map[num_pair.first]-=1;
                    num_map[match_num]-=1;
                }else{
                    ans=false;
                    break;
                }
            }
            if(ans == false) break;   
        }
        return ans;
    }
};