LeetCode 1711 大餐计数[Map] HERODING的LeetCode之路

Posted 2021-08-03 HERODING23

解题思路：
本质是从大餐中找出所有的a + b（根据下标不同，即使值相同也是不同的组合），满足a + b = 2的倍数，那么其实可以转换成 2的倍数 - b = a，所以本题用Map记录a的个数，b为当前遍历到的值，代码如下：

class Solution {
public:
    int countPairs(vector<int>& deliciousness) {
        int MOD = 1e9 + 7;
        int count = 0;
        // 统计2的倍数集合
        vector<int> collection2(22);
        for(int i = 0; i <= 21; i ++) {
            collection2[i] = 1 << i;
        }
        // 记录遍历过的数
        unordered_map<int, int> mp;
        for(int& delicious : deliciousness) {
            for(int num : collection2) {
                // 这里用count比直接判断是否为0用时效率提高了四倍
                if(mp.count(num - delicious)) {
                    count = (count + mp[num - delicious]) % MOD;
                }
            }
            mp[delicious] ++;
        }
        return count;
    }
};