力扣 1711. 大餐计数(中等)

题目

大餐 是指 恰好包含两道不同餐品 的一餐,其美味程度之和等于 2 的幂。
你可以搭配 任意 两道餐品做一顿大餐。
给你一个整数数组 deliciousness ,其中 deliciousness[i] 是第 i​​​​​​​​​​​​​​ 道餐品的美味程度,返回你可以用数组中的餐品做出的不同 大餐 的数量。结果需要对 109 + 7 取余。
注意,只要餐品下标不同,就可以认为是不同的餐品,即便它们的美味程度相同。
示例 1:

输入:deliciousness = [1,3,5,7,9]
输出:4
解释:大餐的美味程度组合为 (1,3)(1,7)(3,5)(7,9) 。
它们各自的美味程度之和分别为 48816 ,都是 2 的幂。

示例 2:

输入:deliciousness = [1,1,1,3,3,3,7]
输出:15
解释:大餐的美味程度组合为 3(1,1)9(1,3) ,和 3(1,7)

提示:

  • 1 < = d e l i c i o u s n e s s . l e n g t h < = 1 0 5 1 <= deliciousness.length <= 10^{5} 1<=deliciousness.length<=105
  • 0 < = d e l i c i o u s n e s s [ i ] < = 2 20 0 <= deliciousness[i] <= 2^{20} 0<=deliciousness[i]<=220

题解

哈希表法
一般而言想到的朴素解法就是双重遍历相加,计算两个元素之和是否等于2的幂,但该解法时间复杂度维 O ( n 2 ) O(n^2) O(n2),超出时间限制。
上面的朴素解法存在同一个元素被重复计算的情况,因此可以使用哈希表来减少重复计算,这个方法很重要。
具体而言,遍历到数组deliciousness中的某个元素时,在哈希表中寻找与当前元素的和等于2的幂的元素个数,然后将当前元素添加进哈希表。由于遍历数组时,哈希表中已有的元素的下标一定小于当前元素的下标,因此任意一对元素之和等于 2 的幂的元素都不会被重复计算。
类似于 a [ i ] + a [ j ] = 2 n , i < j a[i] + a[j] = 2^n, i<j a[i]+a[j]=2n,i<j,开始遍历到 a [ i ] a[i] a[i]时,map中没有 a [ j ] a[j] a[j],将 a [ i ] a[i] a[i]存入map,当遍历到 a [ j ] a[j] a[j]时,就可以将 a [ i ] a[i] a[i]的数量取出,这就得到了一个 a [ j ] a[j] a[j]对应的组合数。当然, a [ j ] a[j] a[j]也要添加进map中。
这样就使用哈希表存储了数组中每个元素出现的次数。
令maxVal表示数组deliciousness中的最大元素,则数组任意两个元素之和都不会超过maxVal×2。令 m a x S u m = m a x V a l × 2 maxSum=maxVal×2 maxSum=maxVal×2,则我们可以从两元素之和 s u m = 2 0 sum=2^0 sum=20开始遍历,对当前元素在map中获取和为sum的另一元素数量,每轮更新 s u m < < = 1 sum <<= 1 sum<<=1,最大为maxSum。
这样就很巧妙,每个元素只需要遍历一次,map中存储每个元素的数量,然后遍历不同的sum,找符合条件的另一个值的数量。

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<>();
        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);  // 将该值加入map中
        }
        return pairs;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值