题目
大餐 是指 恰好包含两道不同餐品 的一餐,其美味程度之和等于 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) 。
提示:
- 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;
}
}