写点什么

【LeetCode】大餐计数 Java 题解

用户头像
HQ数字卡
关注
发布于: 2021 年 07 月 07 日

题目描述

大餐 是指 恰好包含两道不同餐品 的一餐,其美味程度之和等于 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 的幂。
来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/count-good-meals著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
复制代码

思路分析

  • 这个题目题意容易理解,简述成一句话就是求任意两个数之和为 2 的幂的组合数

  • 根据题目实际情况,采用 hashMap 数据结构实现快速查找。

  • 通过遍历 deliciousness, 找到最大的元素,任意 2 个元素之和都小于等于元素的 2 倍,可以简化运算过程,提升效率。

代码

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

总结

  • 上述代码的时间复杂度是 O(n * n), 空间复杂度是 O(n)

  • 坚持每日一题,加油!

发布于: 2021 年 07 月 07 日阅读数: 5
用户头像

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

评论

发布
暂无评论
【LeetCode】大餐计数Java题解