781. 森林中的兔子

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

解题思路

当相同颜色的兔子尽量多时得到的总兔子数量最少，但是相同数量的颜色不能大于该颜色的数值。
例如 answers = [3,3,3,3,3]
这是answers[0],代表还有三只同色兔子，可能为answersrs [1]…[n]中的最多三个。
这是answers[1],代表还有三只同色兔子，可能为answersrs [0],[2]…[n]中的最多三个。
这是answers[2],代表还有三只同色兔子，可能为answersrs [0],[1],[3]…[n]中的最多三个。
…
为方便起见，可以假定前n+1个出现的answers[] =n;为同色
所以假定answers[0] =3 为红色，则第二只 answers= 0，也为红色
第三只 answers= 0，也为红色，则第四只 answers= 0，也为红色。
但是当第四只出现answers= 0时，因为除前四只各自本身加额外三只红色，共四只。
所以当出现第n+2个answers[] =n时只能为另一种颜色。
依次类推。。

代码

class Solution {
    public int numRabbits(int[] answers) {
        HashMap<Integer,Integer> mp = new HashMap<>();
        int num =0;
        for(int i = 0;i<answers.length;i++){
            if(!mp.containsKey(answers[i])){
                mp.put(answers[i],answers[i]);
                num+=answers[i]+1;
            }else{
                if(mp.get(answers[i]) != 0){
                    mp.put(answers[i],mp.get(answers[i])-1);
                }else{
                    mp.put(answers[i],answers[i]);
                    num+=answers[i]+1;
                }
            }
        }
        return num;
    }
}

复杂度分析
时间复杂度:(n),数组只需遍历一次（HashMap用hash值查找）
空间复杂度:(n)