781.森林中的兔子

题目

森林中，每个兔子都有颜色。其中一些兔子（可能是全部）告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。

返回森林中兔子的最少数量。

示例

输入: answers = [1, 1, 2]
输出: 5
解释:
两只回答了 "1" 的兔子可能有相同的颜色，设为红色。
之后回答了 "2" 的兔子不会是红色，否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外，森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。

说明

1. answers 的长度最大为1000。
2. answers[i] 是在 [0, 999] 范围内的整数

题解

1. 同一种颜色的兔子回答一定一样，但是回答一样的兔子不一定是一种颜色
2. 数组中的值可能出现一次或多次
   1. 出现一次，值为others的元素只出现一次：表示除自身外与其颜色相同的兔子的数目。即：该颜色的兔子一种有others+1只
   2. 出现多次，值为others的元素出现了count次：
   3. 我们假设某种颜色的兔子一共有all只，兔子回答答案的数值是others。两者满足：all = others + 1
   4. 但是数组内回答others的兔子数量为count
      1. count <= others + 1：此时表明该颜色的兔子有的没有回答。为了达到【最少的兔子数量】可以假定这count只兔子为同一种颜色
      2. count > others + 1：若一种颜色的兔子全部回答，则count应该为others+1.此时count>others+1表明至少有两种颜色的兔子回答了相同的结果。为了达到【最少兔子数量】，我们尽可能的让每一种兔子的数量达到其上限：others+1

参考代码

class Solution {
    public int numRabbits(int[] answers) {
      Map<Integer, Integer> map = new HashMap<>();
      
      for(int i = 0; i < answers.length; i++) {
        int val = map.getOrDefault(answers[i], 0) + 1;
        map.put(answers[i], val);
      }
      
      int res = 0;
      
      for(int key : map.ketSet()) {
        if(key == 0) {
          res += map.get(key);
        } else if(map.get(key) % (key + 1) == 0) {
          res += map.get(key) / (key + 1) * (key + 1);
        } else {
          res += (map.get(key) / (key + 1) + 1) * (key + 1);
        }
      }
      
    }
}