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

提示：

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

 

解题思路

使用map记录数组A中每个元素的个数，对A进行排序，然后遍历数组A，取key为当前数组元素A[i]，如果key小于0，key除2；否则key乘2，然后查找map中key对应的value是否存在，存在则减一，否则返回false

bool canReorderDoubled(vector<int>& A) {
        int len = A.size();
        if(len == 0) return true;
        map<int,int> mp;
        for(int i=0;i<len;i++){
            if(mp[A[i]] == 0) mp[A[i]] = 1;
            else mp[A[i]] += 1;
        }
        sort(A.begin(),A.end());
        for(int i=0;i<len;i++){
            int key = A[i];
            if(mp[key] > 0){
                if(key < 0) key /= 2;
                else key *= 2;
                if(mp[key] == 0) return false;
                mp[key]--;
                mp[A[i]]--;
            }
        }
        return true;