781. Rabbits in Forest

weixin_30675247

于 2018-09-02 10:12:00 发布

阅读量56

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/learning-c/p/9572848.html

本文介绍了一种用于统计兔子数量的算法实现。通过解析输入列表answers中的数据，利用字典记录每种答案出现的次数，并根据特定规则计算出最少需要的兔子数量。

class Solution(object):
    def numRabbits(self, answers):
        """
        :type answers: List[int]
        :rtype: int
        """
        dict={}
        for x in answers:
            if x in dict.keys():
                dict[x]=dict[x]+1
            else:
                dict[x]=1
        cnt=0
        for k in dict.keys():
            v=dict[k]
            if v%(k+1)==0:
                cnt+=v/(k+1)*(k+1)
            else:
                cnt+=v/(k+1)*(k+1)+(k+1)
        return cnt

转载于:https://www.cnblogs.com/learning-c/p/9572848.html