森林中,每个兔子都有颜色。其中一些兔子(可能是全部)告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。
返回森林中兔子的最少数量。
示例:
输入: answers = [1, 1, 2]
输出: 5
解释:
两只回答了 "1" 的兔子可能有相同的颜色,设为红色。
之后回答了 "2" 的兔子不会是红色,否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。
输入: answers = [10, 10, 10]
输出: 11
输入: answers = []
输出: 0
哈希的代码很简单,不过这道题的规则的理解有难度。
public int NumRabbits(int[] answers)
{
/*
* 依据诚实的兔子们的回答,推断出最少有多少只兔子
* 思路:
* 1.每个回答都是诚实的,那么回答数量相同的,一定满足下面的推断
* 1.1 回答的数量+1=此种颜色兔子的数量
* 1.2 答案相同的兔子数量 > 此种颜色兔子的数量,那么说明有多个颜色的兔子数量相同
* 2.所以需要先对兔子的回答做汇总,然后对每种答案求解
* 2.1 (答案相同的兔子数量 / 此种颜色兔子的数量)向上取整,然后乘以此种颜色兔子的数量
*
* 时间复杂度:O(n),遍历数组的每一项
* 空间复杂度:O(n),最坏情况下,每个兔子的回答都不同
*/
var colorGroupDic = new Dictionary<int, int>(answers.Length);
foreach(var answerItem in answers)
{
if (!colorGroupDic.ContainsKey(answerItem)) colorGroupDic[answerItem] = 0;
colorGroupDic[answerItem]++;
}
var forReturn = 0;
foreach (var colorItem in colorGroupDic)
{
var colorTotalNum = colorItem.Key + 1;
forReturn += (int)Math.Ceiling(1.0 * colorItem.Value / colorTotalNum) * colorTotalNum;
}
return forReturn;
}