1. 题目描述
森林中有未知数量的兔子。提问其中若干只兔子 "还有多少只兔子与你(指被提问的兔子)颜色相同?" ,将答案收集到一个整数数组 answers
中,其中 answers[i]
是第 i
只兔子的回答。
给你数组 answers
,返回森林中兔子的最少数量。
示例 1:
输入:answers = [1,1,2]
输出:5
解释:
两只回答了 "1" 的兔子可能有相同的颜色,设为红色。
之后回答了 "2" 的兔子不会是红色,否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5 只:3 只回答的和 2 只没有回答的。
示例二:
输入:answers = [10,10,10]
输出:11
提示:
1 <= answers.length <= 1000
0 <= answers[i] < 1000
2. 分析
核心思路:贪心+哈希表聚类
分析:
要求最少数量,我们可以贪心地认为,每个回答相同数量的兔子,都是同一种颜色。
通过哈希表,聚合相同回答的兔子,比如回答10,那么应该至少有11只相同颜色的兔子。
特殊地:如果回答10,但是答案中出现了10个以上10。那么超出的数量应该划分至下一组颜色。
3. 代码实现
注:本文由于值域有限[0-1000),故使用数组作记忆化存储。
使用数组的好处:
预分配内存空间,后续无需触发二次扩容和数据再分配。
可以直接基于索引访问数据,无需再调用hash函数计算hash,和寻址数据,性能稍优。
// 贪心:要分析最少数量。我们可以贪心的认为回答相同数量的兔子,都有相同的颜色
// 通过哈希表或者值域数组,聚合答案
func numRabbits(answers []int) int {
ans := 0
hset := [1000]int{}
for _, v := range answers {
// 记忆化,同组归并
if hset[v] > 0 && hset[v] < v+1 {
// 同组元素累计,超过v+1(同组元素最大数量),则重置
hset[v]++
continue
}
// 重置:回答没有出现过。或者同组的回答超出了限制,需要重置
hset[v] = 1
ans += v+1
}
return ans
}
评论