[数据结构与算法]leetcode 1711. 大餐计数

大餐 是指 恰好包含两道不同餐品 的一餐，其美味程度之和等于 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) 。

解题思路

• 因为deliciousness[i]最大就是2的20次方，因此美味程度之和只有1到2的21次幂，共21种选择
• 使用map记录每个元素出现的次数
• 做出大餐的条件：deliciousness[i]+deliciousness[x]=2的n次幂，变形得deliciousness[x]=2的n次幂-deliciousness[i]，因此我们可以遍历2的n次幂，通过map找出deliciousness[x]的出现次数，即是做出大餐的次数

代码

class Solution {
    public int countPairs(int[] deliciousness) {

        int two=1,mod=1000000007;
        ArrayList<Integer> list = new ArrayList<>();

        for (int i=0;i<=21;i++)
        list.add(two<<i);
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(deliciousness[0],1);
        Long res= 0L;
        for (int i=1;i<deliciousness.length;i++)
        {
            for (Integer integer : list) {
                if(map.containsKey(integer-deliciousness[i]))
                  res=(res+map.get(integer-deliciousness[i]))%mod;
            }
            map.put(deliciousness[i],map.getOrDefault(deliciousness[i],0)+1);
        }
        return (int)(res%(10e9+7));

    }
}