1710 - 大餐计数 - 哈希查找 - 枚举 - 排序 -双指针

最新推荐文章于 2024-05-24 14:56:56 发布

闪电彬彬

最新推荐文章于 2024-05-24 14:56:56 发布

阅读量113

点赞数

分类专栏： leetcode 文章标签：哈希枚举类排序算法

本文链接：https://blog.csdn.net/chenbb1989/article/details/112151100

版权

leetcode 专栏收录该内容

76 篇文章 1 订阅

订阅专栏

欢迎关注更多精彩
关注我，学习常用算法与数据结构，一题多解，降维打击。

文章目录

题目描述

[1710] 大餐计数

https://leetcode-cn.com/problems/count-good-meals/

大餐是指恰好包含两道不同餐品的一餐，其美味程度之和等于 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 <= deliciousness.length <= 10^5
0 <= deliciousness[i] <= 2^20

题目剖析&信息挖掘

此题使用哈希与枚举的思想，注意到deliciousness[i]说明数组里2数之和最大是2^21

可以用哈希实现，也可以用双指针法实现

解题思路

方法一哈希+枚举

分析

把题目简化一下，问题数组中2数加起来等2的x次方的方法数是多少。

之前类似的题目有，K 和数对的最大数目(https://leetcode-cn.com/problems/max-number-of-k-sum-pairs/) ，两数之和（https://leetcode-cn.com/problems/two-sum/）

可以用哈希法去解决，复杂度是O(n)

现在根据题目的条件，x从原来的1个值变成了x = [0, 21]

相当于枚举x, 然后做两数之和，复杂度是22 * O(n) = 10^5*20 = 10^6, 时间上差刚好够。

思路

func countPairs(deliciousness []int) int {
	cntMap  := make(map[int]int, len(deliciousness)) // 用来存储之前数的统计个数
	sum :=0
	for _, v := range deliciousness {

		for j:=uint(0);j<22; j++  {
			left := (1<<j)-v
			sum = (cntMap[left]+sum)%MOD // 查找另一半
		}
		
		cntMap[v]++
	}
	return sum
}

注意

要取模
先统计，再把当前数字加到map中。

知识点

哈希查找
二进制
枚举

复杂度

时间复杂度：O(n)
空间复杂度：O(n)

参考

代码实现

const MOD = int(1e9)+7

func countPairs(deliciousness []int) int {
	cntMap  := make(map[int]int, len(deliciousness))
	sum :=0
	for _, v := range deliciousness {

		for j:=uint(0);j<22; j++  {
			if (1<<j)< v {
				continue
			}
			left := (1<<j)-v
			if cnt, ok:=cntMap[left];ok && cnt>0 {
				sum = (cnt+sum)%MOD
			}
		}
		
		cntMap[v]++
	}
	return sum
}