781. 森林中的兔子 (Python 实现)

题目：

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

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

输入: answers = [10, 10, 10]
输出: 11

输入: answers = []
输出: 0

解题思路：
如果有多个兔子报相同的数字x，那么报x的这个兔子的颜色最多有x+1只，如果报x的兔子大于x+1只那么多的兔子并不属于这种颜色。
对于[1, 1, 1, 1]那么该列子中假设第一个为红色，第二个也为红色，那么第三个肯定不能为红色了，可以为蓝色，第四个也要为蓝色。
即前两个为一类，后两个就为另一类了，相同数字1的兔子一共有4//（1+1）= 2 类颜色四只兔子

对于[1, 1, 1, 1，1]那么该列子中假设第一个为红色，第二个也为红色，那么第三个肯定不能为红色了，可以为蓝色，第四个也要为蓝色，但是还有第五只兔子也报1，那么这个第五只兔子就不能为红色和蓝色了，可以为白色。
即前两个为一类，后两个就为另一类了，第五个单独为一类，相同数字1的兔子一共有5//（1+1） + 1= 3 类颜色六只兔子。因为第五只兔子报告还有之一兔子和他统一颜色，只不过那只兔子并没有说话。但根据第五只兔子的话表明了还有一只兔子的存在。

对于[2，2，2，2]也是有六只兔子

代码：

class Solution:
    def numRabbits(self, answers: List[int]) -> int:
        import math
        ans=set(answers)
        res=0
        for x in ans:
            res+=ceil(answers.count(x)/(x+1))*(x+1)
        return res