题目直达

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. 分析

核心思路：贪心+哈希表聚类

分析：

1. 要求最少数量，我们可以贪心地认为，每个回答相同数量的兔子，都是同一种颜色。

2. 通过哈希表，聚合相同回答的兔子，比如回答10，那么应该至少有11只相同颜色的兔子。

3. 特殊地：如果回答10，但是答案中出现了10个以上10。那么超出的数量应该划分至下一组颜色。

3. 代码实现

注：本文由于值域有限[0-1000)，故使用数组作记忆化存储。

使用数组的好处：

1. 预分配内存空间，后续无需触发二次扩容和数据再分配。

2. 可以直接基于索引访问数据，无需再调用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
}