【中等】大餐计数

题目描述

大餐 是指 恰好包含两道不同餐品 的一餐，其美味程度之和等于 2 的幂。

你可以搭配 任意 两道餐品做一顿大餐。

给你一个整数数组 deliciousness ，其中 deliciousness[i] 是第 i​​​​​​​​​​​​​​ 道餐品的美味程度，返回你可以用数组中的餐品做出的不同 大餐 的数量。结果需要对 $10^9+7$ 取余。

注意，只要餐品下标不同，就可以认为是不同的餐品，即便它们的美味程度相同。

示例 1：
输入：deliciousness = [1,3,5,7,9]
输出：4
解释：大餐的美味程度组合为 (1,3) 、(1,7) 、(3,5) 和 (7,9) 。
它们各自的美味程度之和分别为 4 、8 、8 和 16 ，都是 2 的幂。

示例 2：
输入：deliciousness = [1,1,1,3,3,3,7]
输出：15
解释：大餐的美味程度组合为 3 种 (1,1) ，9 种 (1,3) ，和 3 种 (1,7) 。

提示：
1 <= deliciousness.length <= $10^5$
0 <= deliciousness[i] <= $2^{20}$

解题思路

由于0 <= deliciousness[i] <= $2^{20}$，所以大餐的美味程度一定在[0,$2^{21}$]之间，即$2^0，2^1，...，2^{21}$共22种。

将deliciousness中每个美味程度出现的次数存储到map中。遍历该map，对于每个美味程度，计算与其匹配的美味程度（共22个）。如果该匹配项也在map中，再判断匹配项与源项是否一致，如果一致，按$\frac{n(n-1)}{2}$计算，如果不一致，按$n\times m$计算。其中$n$=map[源项]，$m$=map[匹配项]。

class Solution {
public:
    int countPairs(vector<int>& deliciousness) {
        unordered_map<int, int> nums;
        long ans1=0,ans2=0;
        for(auto it:deliciousness){
            if(nums.find(it)!=nums.end()){
                nums[it]++;
            }else{
                nums[it]=1;
            }
        }
        for(auto it:nums){
            vector<int> res=get_mi(it.first);
            for(int ite:res){
                if(nums.find(ite)!=nums.end()){
                    if(it.first!=ite) ans2=add(ans2,it.second*nums[ite],2);
                    else ans1=add(ans1,((long)it.second*(it.second-1)/2),1);
                }
            }
        }
        return (ans1+ans2/2)%(int)(1e9+7);
    }

    long add(long a,long b,int d){
        int c=1e9+7;
        if(a+b>d*c) return a+b-d*c;
        else return a+b;
    }

    vector<int> get_mi(int a){
        vector<int> res;
        int b=1;
        for(int i=0;i<=21;i++){
            res.push_back(b-a);
            b<<=1;
        }
        return res;
    }
};