Create maximum-number-of-balls-in-a-box.py

Author: kamyu104

Python/maximum-number-of-balls-in-a-box.py

@@ -0,0 +1,18 @@
+# Time:  O(nlogm)
+# Space: O(logm)
+
+import collections
+import itertools
+
+
+class Solution(object):
+    def countBalls(self, lowLimit, highLimit):
+        """
+        :type lowLimit: int
+        :type highLimit: int
+        :rtype: int
+        """
+        count = collections.Counter()
+        for i in xrange(lowLimit, highLimit+1):
+            count[sum(itertools.imap(int, str(i)))] += 1
+        return max(count.itervalues())