LeetCode刷题笔记-森林中的兔子_喜欢的兔子 leetcode-CSDN博客

本文链接：https://blog.csdn.net/m0_51051154/article/details/115529984

森林中的兔子

森林中，每个兔子都有颜色。其中一些兔子（可能是全部）告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 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;
		  
    }