2021-04-04 leetcode 781. 森林中的兔子

题目描述

森林中，每个兔子都有颜色。其中一些兔子（可能是全部）告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 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] 范围内的整数。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rabbits-in-forest

解题思路

一开始读了本题,其实应该对所有的回答先进行排序,如果想要最少兔子数量,就应该把回答数量一样的 兔子 整合到一种颜色,比如例1 中的 1,1,2,把前两者 认为是同一颜色的兔子,这样就可以,使得兔子数量最少,将 回答为1的兔子认为 白色,也就是白色兔子 最少可以有两只,回答为2的兔子 认为是灰色兔子,这样的话 除了他自己之外还有两只灰色兔子,也就是总共有三只灰色兔子.这样的话 就是最少可以为 5只兔子.
如果每一个回答的兔子各自颜色都不同,数量就是每个回答的数量各自加1,因为还要加上他自己.
最理想的情况就是兔子回答数组 anwsers 经过排序之后,每一组回答相同的兔子都是颜色相同的,但是 假设 如果出现 0,1,1,1,2这种情况, 假设回答为0 的兔子颜色为A只有他自己一只, 回答为1 的颜色为B,B类兔子应该如果都合起来应该是 2只,但是有三只兔子都回答说 和他同一颜色的兔子只有一只,最多只能两只回答为1的兔子是同一颜色,另外一只回答为1 的兔子颜色只能为C,
C类兔子的数量最少为2,回答为 3的兔子颜色为D,D类兔子最少数量为3,所以当回答相同的兔子排在一起时,还需要判断 兔子的数量和统一回答的个数之前的大小关系.

提交结果

代码实现

class Solution {
    public int numRabbits(int[] answers) {
        int sum = 0;
        Arrays.sort(answers);
        int i = 0;
        while (i<answers.length-1){
        //将相同回答的兔子整理为一组
            if (answers[i]==answers[i+1]){
                int start = i;
                int end = i;
                while (i<answers.length-1 && answers[i]==answers[i+1]){
                    end++;
                    i++;
                }
                sum += getNum(start,end,answers);
            }else {
                sum += answers[i]+1;
            }
            i++;
        }
        int len = answers.length;
        //如果最后一个为单独一组,上面遍历时不能计算最后一个兔子数量,所以单独再计算
        if (answers[len-2]!=answers[len-1]){
            sum += answers[len-1]+1;
        }
        return sum;
    }
    //递归获取相同回答的一组兔子的最少个数
    public int getNum(int start,int end,int[] anwsers){
        int sum = 0;
        if (end-start<=anwsers[start]){
            sum = anwsers[start]+1;
        }else {
            sum = anwsers[start]+1;
            sum += getNum(start+anwsers[start]+1,end,anwsers);
        }
        return sum;
    }
}

标签：781,04,int,sum,回答,兔子,start,answers,2021
来源： https://blog.csdn.net/qq_40600414/article/details/115432564