1711. 大餐计数

Posted 2021-08-06 ai52learn

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

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

给你一个整数数组 deliciousness ，其中 deliciousness[i] 是第 i​​​​​​​ 道餐品的美味程度，返回你可以用数组中的餐品做出的不同 大餐 的数量。结果需要对 109 + 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 <= 105
0 <= deliciousness[i] <= 220

class Solution {
public:
    int countPairs(vector<int>& deliciousness) {
        const int INF = 1e9+7;
        int mp[1048576*2+5]={0};
        for(auto t:deliciousness)
            mp[t]++;
        int ret = 0;
        int t = 1;
        for(int i =0;i<22;i++)
        {
            for(int j = 0;j<t/2;j++)
            {
                ret =(ret+(long long)mp[j]*mp[t-j]%INF)%INF;
            }
            if(t%2==0)
            {
                ret = (ret+(long long)mp[t/2]*(mp[t/2]-1)/2%INF)%INF;
            }
            else
            {
                ret =(ret+(long long)mp[t/2]*mp[t-t/2]%INF)%INF;
            }
            t*=2;
        }
        return ret;
    }
};