Update probability-of-a-two-boxes-having-the-same-number-of-distinct-balls.py

Author: kamyu104

Python/probability-of-a-two-boxes-having-the-same-number-of-distinct-balls.py

@@ -12,11 +12,10 @@ def getProbability(self, balls):
         """
         def nCrs(n):  # Time: O(n), Space: O(1)
             c = 1
-            yield c
-            for k in xrange(1, n+1):
-                c *= n-k+1
-                c //= k
+            for k in xrange(n+1):
                 yield c
+                c *= n-(k+1)+1
+                c //= k+1
 
         def nCr(n, r):  # Time: O(n), Space: O(1)
             if n-r < r: