解题思路:
本质是从大餐中找出所有的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;
}
};