力扣 1711. 大餐计数

题目来源： https://leetcode-cn.com/problems/count-good-meals/

大致题意：
给定一个整数数组，然后算出数组中元素两两相加的和等于2的幂的元素对一共有多少。结果需要对10^9+7取余。

思路

统计出最大值。然后可知：

• 其他所有元素对的和 <= 最大值 × 2

于是我们遍历数组，同时用一个HashMap存在当前遍历位置前出现的数字以及数字对应的次数。
然后对于当前每个数字元素 num，从 1 开始，遍历到 最大值*2，假设遍历值为sum，每次使用 sum 减去 num，然后查找HashMap中该差出现的次数，次数即代表在当前数字元素num之前，数组中已经有多少个元素可以和num组成2的幂。然后将结果加上次数，对10^9+7取余，再将当前num统计入HashMap。

代码：

public int countPairs(int[] deliciousness) {
        int pair = 0;
        int maxVal = 0;
        int mod = 1000000007;
        for (int i : deliciousness) { // 获取最大元素
            maxVal = Math.max(maxVal, i);
        }
        
        Map<Integer, Integer> numMap = new HashMap<Integer, Integer>(); // 存储deliciosness数组中每个数出现次数
        for (int num : deliciousness) {

            for (int sum = 1; sum <= maxVal*2; sum <<= 1) {
                int val = sum - num; // 获取和当前元素值相加和为2的幂的数
                if (val > maxVal)
                    break;
                int count = numMap.getOrDefault(val, 0); // 获取该数已经出现的次数，若无则为0
                // 次数 代表在当前num之前deliciouness数组已有 多少数 可以和 当前数 组成大餐
                pair = (pair + count) % mod; // 加上次数取余
            }
            numMap.put(num, numMap.getOrDefault(num, 0) + 1); // 将当前数放入Map，即统计当前数已经出现多少次
        }

        return pair;
    }