一、题目
森林中有未知数量的兔子。提问其中若干只兔子 "还有多少只兔子与你(指被提问的兔子)颜色相同?" ,将答案收集到一个整数数组 answers 中,其中 answers[i] 是第 i 只兔子的回答。
给你数组 answers ,返回森林中兔子的最少数量。
示例 1:
输入:answers = [1,1,2]
输出:5
解释:
两只回答了 "1" 的兔子可能有相同的颜色,设为红色。
之后回答了 "2" 的兔子不会是红色,否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5 只:3 只回答的和 2 只没有回答的。
示例 2:
输入:answers = [10,10,10]
输出:11
提示:
1 <= answers.length <= 1000
0 <= answers[i] < 1000
二、代码
class Solution {
public int numRabbits(int[] answers) {
// 过滤无效参数
if (answers == null || answers.length == 0) {
return 0;
}
// 排序,把说的数字相同的人排列到一起
Arrays.sort(answers);
// 当前统计的是说num的人有多少个
int num = answers[0];
// 统计当前这一组有多少人
int cnt = 1;
int ans = 0;
// 看一下说的数字相同的人能划分出几组,再用组数乘以num+1就是这一组能确定的最少人数
for (int i = 1; i < answers.length; i++) {
// 当answers[i] != num时,说明已经遍历完了一组,开始结算这一组,并且开始遍历说下一种数字的一组
if (answers[i] != num) {
// 结算说num的人的这一组能确定的最少人数
ans += (cnt + (num + 1 - 1)) / (num + 1) * (num + 1);
// 后面开始找说answers[i]这个数字的人有多少个,更新num的值
num = answers[i];
// cnt初始化为1,开始找说answers[i]这个数字的人有多少个
cnt = 1;
// 遍历看说num这个数字的人有多少个,cnt++
} else {
cnt++;
}
}
// 还需要再加上最后一轮计算的,上面循环的最后一轮计算的还没有结算进ans中
return ans + (cnt + (num + 1 - 1)) / (num + 1) * (num + 1);
}
}
三、解题思路
是一道贪心题,将说的一样的人划分成一大组,让他们自己在组内消化,计算可能的最少人数。