森林中的兔子---Leetcode练习

思路很重要

题目描述

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

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

思路

建立一个字典，保存每次相同回答的次数，比如answers=[1,0,1,1,0,0]，新建字典为dt=[1:3,0:3]，因为回答“1”表示还有一只兔子与自己颜色相同，那么该颜色的兔子应该是两只，然而字典中“1”对应的值是3，说明有三只兔子表示还有一只兔子颜色与自己相同，那么最少的情况是有四只兔子，最多的情况是六只兔子。

如果你不是一个Python使用者也可以创建一个二维数组来记录不同回答的兔子的数量。

题目要求求最少的情况，那么我们只要用回答n的兔子的数量m除以n+1向上取整，然后再乘以n+1就行了。按照这样的做法，遍历一遍不同的回答就可以求出最少的兔子的数量了。

代码

class Solution:
    def numRabbits(self, answers: List[int]) -> int:
    	#新建一个字典
        end=dict()
        #记录最少兔子的数量
        nums=0
        for i in range(len(answers)):
            if answers[i] not in end:
                end[answers[i]]=1
            else:
                end[answers[i]]+=1
        for key in end.keys():
            if end[key]%(key+1)==0:
                nums+=end[key]
            else:
                nums+=((end[key]//(key+1))+1)*(key+1)
        return nums

编码不易，有帮助的话点个赞支持一下吧