aboutsummaryrefslogtreecommitdiff
path: root/rabbits_in_forest.py
diff options
context:
space:
mode:
Diffstat (limited to 'rabbits_in_forest.py')
-rw-r--r--rabbits_in_forest.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/rabbits_in_forest.py b/rabbits_in_forest.py
new file mode 100644
index 0000000..7d91e99
--- /dev/null
+++ b/rabbits_in_forest.py
@@ -0,0 +1,30 @@
+# 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]))