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

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

阅读量：717 次

发布时间：2019-03-21

本文共 1861 字，大约阅读时间需要 6 分钟。

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

文章目录

题目描述

[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<

注意

要取模

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

知识点

哈希查找

二进制

枚举

复杂度

时间复杂度：O(n)

空间复杂度：O(n)

参考

代码实现

const MOD = int(1e9)+7func 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<
   
    < v {
   				continue			}			left := (1<
    
     0 {
   				sum = (cnt+sum)%MOD			}		}				cntMap[v]++	}	return sum}

方法二枚举+排序+双指针

可参考关注公众号回复 lt1697 ，自行实现

注意

要处理数字重复的情况