森林中的兔子
森林中,每个兔子都有颜色。其中一些兔子(可能是全部)告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。
返回森林中兔子的最少数量。
示例:
输入: answers = [1, 1, 2]
输出: 5
解释: 两只回答了 “1” 的兔子可能有相同的颜色,设为红色。
之后回答了 “2” 的兔子不会是红色,否则他们的回答会相互矛盾。 设回答了 “2” 的兔子为蓝色。 此外,森林中还应有另外 2
只蓝色兔子的回答没有包含在数组中。 因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。
输入: answers = [10, 10, 10]
输出: 11
输入: answers = []
输出: 0
说明:
answers 的长度最大为1000。
answers[i] 是在 [0, 999] 范围内的整数。
我的答案
思路在代码注释里面
public static int numRabbits(int[] answers) {
//1.遍历一遍数组,
//回答相同的: 数字可能是同一种颜色(最小的可能)
// 在1下同一种颜色下:答案相同数量 <=(回答数字+1)
//回答不同的: 颜色+1
int len=answers.length;
int i=0 ,j=0,count=0,sum=0;//i定位新元素,j找和新元素相同的元素,count记录出现次数,sum记录兔子数量
while(i<len) {
if(answers[i]==-1) { i++; continue;}//已经找过的重复的区域
j=i+1;count=0;//每一轮需要回空
//遍历找相同
while(j<len) {
if(answers[j] == answers[i]) { answers[j]= -1; count++; }//找到了就标记为-1
j++;
}
if(answers[i]==0) {sum+=(count+1); i++;continue;}//如果回答是0,有几个0就有几个不同颜色的兔子
//用count 算出最小的颜色种数
if(count==0) { //只出现一次的
sum+= answers[i]+1;
} else if(count<= answers[i] ) { //可以认为是同一种颜色的不同兔子回答的
sum+=answers[i]+1;
} else{ //不同颜色的相同答案的兔子
int zhi=( count/(answers[i]+1) +1 );//几组不同颜色的兔子
for(int k=0;k<zhi;k++) {
sum+=answers[i]+1;
}
}
i++;
}
return sum;
}
时间复杂度O(n^2)…是我写的…垃圾算法