LeetCode 954. 二倍数对数组（map计数）

1. 题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/array-of-doubled-pairs
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• map计数，检查 每个元素的2倍 或者 偶数的一半 是否存在

class Solution {
public:
    bool canReorderDoubled(vector<int>& A) {
        map<int,int> m;
        for(int a : A)
            m[a]++;
        int count = 0, a, c;
        for(auto it = m.begin(); it != m.end(); ++it)
        {
            a = it->first;
            if(m.count(2*a) && m[2*a] > 0)//两倍的数存在
            {
                if(a != 0)
                    c = min(m[2*a], m[a]);
                else// a == 0
                    c = m[a]/2;
                count += 2*c;
                m[a] -= c;
                m[2*a] -= c;
            }
            if(a%2==0 && m.count(a/2) && m[a/2] > 0)//是偶数，且一半存在
            {
                if(a != 0)
                    c = min(m[a/2], m[a]);
                else
                    c = m[a]/2;
                count += 2*c;
                m[a] -= c;
                m[a/2] -= c;
            }
        }
        return count == A.size();
    }
};

276 ms 53.7 MB

or
下面的写法思路更清晰

class Solution {
public:
    bool canReorderDoubled(vector<int>& A) {
        map<int,int> m;
        for(int a : A)
            m[a]++;
        int count = 0, a;
        for(auto it = m.begin(); it != m.end(); ++it)
        {
            a = it->first;
            if(it->second == 0)
                continue;
            if(a < 0)
            {
                if(a&1)//负奇数，不能匹配
                    return false;
                m[a/2] -= m[a];
                if(m[a/2] < 0)//个数不够不能匹配
                    return false;
            }
            else if(a == 0)
            {
                if(m[a]&1)//0 奇数个，不能匹配
                    return false;
            }
            else // (a > 0)
            {
                m[2*a] -= m[a];
                if(m[2*a] < 0)//个数不够不能匹配
                    return false;
            }
        }
        return true;
    }
};

236 ms 54 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！