LeetCode知识点总结 - 781

jeeeee_no

于 2024-01-14 16:51:43 发布

阅读量426

点赞数 6

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_59773145/article/details/135584935

版权

LeetCode 专栏收录该内容

417 篇文章

订阅专栏

文章讲述了如何使用贪婪算法解决LeetCode中的问题781，即在森林中有未知数量的兔子，询问它们的颜色分布后，求最少可能的兔子数量。通过Counter计算不同颜色的兔子数量，然后利用答案数组计算组合，得出最少兔子总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LeetCode 781. Rabbits in Forest

考点	难度
Greedy	Medium

题目

There is a forest with an unknown number of rabbits. We asked n rabbits “How many rabbits have the same color as you?” and collected the answers in an integer array answers where answers[i] is the answer of the ith rabbit.

Given the array answers, return the minimum number of rabbits that could be in the forest.

思路

如果x+1只兔子有同一种颜色，最多有x+1只兔子回答x
如果n % (x + 1) == 0，一共有n / (x + 1)组x + 1
如果不为0，一共有n / (x + 1) + 1组x + 1

答案

class Solution(object):
    def numRabbits(self, answers):
        c = collections.Counter(answers)
        return sum((c[i] + i) / (i + 1) * (i + 1) for i in c)