-
题目链接 https://leetcode-cn.com/problems/rabbits-in-forest/comments/
-
题目描述
-
森林中,每个兔子都有颜色。其中一些兔子(可能是全部)告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在
answers
数组里。返回森林中兔子的最少数量。
-
示例: 输入: answers = [1, 1, 2] 输出: 5 解释: 两只回答了 "1" 的兔子可能有相同的颜色,设为红色。 之后回答了 "2" 的兔子不会是红色,否则他们的回答会相互矛盾。 设回答了 "2" 的兔子为蓝色。 此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。 因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。 输入: answers = [10, 10, 10] 输出: 11 输入: answers = [] 输出: 0
-
-
解题思路
- 根据题设,为了保证最少数量,我们假设回答的兔子只有一个。也就是说answers[i]表示某种颜色下的兔子为answers[i]+1个。并且显然答案为answers[i]并且同样颜色的最多为answers[i]。我们可以用map来做上述统计,key是answers[i], value是接下来answers中答案为answers[i]的个数。对answers遍历即可。当answers[i]不存在,或者answers[i]为0时,计数加一
-
代码
- 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
- pyhon
leetcode 781. 森林中的兔子
最新推荐文章于 2021-04-04 19:41:09 发布