【坚持每日一题7.10】1711. 大餐计数

最新推荐文章于 2024-07-30 22:30:00 发布

程序员小2

最新推荐文章于 2024-07-30 22:30:00 发布

阅读量82

点赞数 1

分类专栏： LeetCode

本文链接：https://blog.csdn.net/kangbin825/article/details/118559378

版权

LeetCode 专栏收录该内容

655 篇文章 4 订阅

订阅专栏

该博客介绍了如何解决一个数学问题，即找出一组整数中可以形成美味程度之和为2的幂的大餐组合数量。通过遍历整数数组并使用哈希映射存储已计算过的数值，可以有效地计算出所有可能的组合。示例展示了对于给定的美味程度数组，如何计算出大餐的组合计数，并对结果进行模10^9+7运算以避免溢出。

摘要由CSDN通过智能技术生成

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

java代码：

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