LeetCode1711-大餐计数

题目

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

解法

使用正常的n^2方法肯定会超时
所以首先限制最大的2的幂指数的大小，设数组中的最大的值为max，则2*max >= 最大的2的幂指数然后进行运算。
遍历到数组 deliciousness 中的某个元素时，在哈希表中寻找与当前元素的和等于2 的幂的元素个数，然后用当前元素更新哈希表。由于遍历数组时，哈希表中已有的元素的下标一定小于当前元素的下标，因此任意一对元素之和等于 2 的幂的元素都不会被重复计算。

class Solution {
    public int countPairs(int[] deliciousness) {
        int n = deliciousness.length;
        int MOD = 1000000007;
        int res = 0;
        int max = 0;
        for(int i = 0 ; i < n ; ++i) {
            max = Math.max(max, deliciousness[i]);
        }
        max *= 2;
        HashMap<Integer, Integer> map = new HashMap<>();
        for(int i = 0 ; i < n ; ++i) {
            int val = deliciousness[i];
            for(int cur = 1; cur <= max; cur = cur << 1) {
                int count = map.getOrDefault(cur - val, 0);
                res = (res + count) % MOD;
            }
            map.put(val, map.getOrDefault(val, 0) + 1);
        }
        return res;
    }
}