题目描述
森林中,每个兔子都有颜色。其中一些兔子(可能是全部)告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。
返回森林中兔子的最少数量。
示例
输入: answers = [1, 1, 2]
输出: 5
解释:
两只回答了 "1" 的兔子可能有相同的颜色,设为红色。
之后回答了 "2" 的兔子不会是红色,否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。
输入:answers = [10, 10, 10]
输出:11
输入:answers = []
输出:0
解题思路:
1.乍一看没有思路,所以我试着多举了几组数据:
然后发现,大概思路应该是贪心,就是把所有回答相同的兔子归为同一种颜色,如图所示,[2,2,3,3,4]输出应该是12.
2.问题来了,如果回答相同数量的兔子特别多怎么办呢?
可以发现,如果有四只兔子都回答2的时候,那么数量应该是6,故相同类型的兔子数量应该是3的倍数,依次类推:
所以我们用一个hash表存储回答相同答案的兔子的个数,例如hash[2] = 3意思是回答2的兔子有3只。
然后对每一类最少兔子数量进行计算,有x.second / (x.first + 1) (上取整)这么多群,我们把其化为下取整, 即 a / b(上取整) = (a + b - 1) / b(下取整)
故这里t = (x.second + x.first) / (x.first + 1); 例如hash[5] = 6, 即 t = (5 + 2) / 3 = 2群, 2 * 3 = 6只兔子
3.这样这题基本上就解决了,代码如下:
class Solution {
public:
int numRabbits(vector<int>& answers) {
unordered_map<int, int> hash; //记录回答相同数量的兔子个数,例如:hash[2] = 3意思是回答2的兔子有三只
int n = answers.size();
if(n == 0) return {}; //特判0的情况
for(int i = 0; i < n; i++) hash[answers[i]]++; //统计兔子数目
int res = 0;
for(auto x : hash){
int t = (x.second + x.first) / (x.first + 1);
res += t * (x.first + 1);
}
return res;
}
};