leetcode 781. 森林中的兔子

1. 题目链接 https://leetcode-cn.com/problems/rabbits-in-forest/comments/

2. 题目描述

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

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

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

3. 解题思路

   1. 根据题设，为了保证最少数量，我们假设回答的兔子只有一个。也就是说answers[i]表示某种颜色下的兔子为answers[i]+1个。并且显然答案为answers[i]并且同样颜色的最多为answers[i]。我们可以用map来做上述统计，key是answers[i]， value是接下来answers中答案为answers[i]的个数。对answers遍历即可。当answers[i]不存在，或者answers[i]为0时，计数加一

4. 代码

   1. pyhon

      class Solution:
          def numRabbits(self, answers: List[int]) -> int:
              d = {}
              res =  0
              for ans in answers:
                  if ans in d and d[ans]:
                      d[ans] -= 1
                  else:
                      d[ans] = ans
                      res += ans + 1
              return res