rabbits_in_forest.py (840B)
| # 781. Rabbits in Forest
from collections import Counter
"""
there is a forest with an unknown number of rabbits. we asked 'n' rabbits
"how many rabbithave the same colour as you?" and collected the answers in an
integer array 'answers' where 'answers[i]' is the answer of the i'th rabbit.
given the array 'answers', return the minimum number of rabbits that could be
in the forest.
"""
class Solution(object):
def numRabbits(self, answers):
"""
:type answers: List[int]
:rtype: int
"""
ans, freq = 0, Counter()
for i in answers:
if freq[i] % (i + 1) == 0:
ans += i + 1
freq[i] += 1
return ans
if __name__ == "__main__":
obj = Solution()
print(obj.numRabbits(answers=[1, 1, 2]))
print(obj.numRabbits(answers=[10, 10, 10]))
|
