Update count-the-number-of-substrings-with-dominant-ones.py

Author: kamyu104

Python/count-the-number-of-substrings-with-dominant-ones.py

@@ -15,8 +15,8 @@ def numberOfSubstrings(self, s):
             if idxs[curr] == i:
                 curr += 1
             for c in xrange(min(int((-1+(1+4*(i+1))**0.5)/2)+1, curr)):  # since c^2 <= (i+1)-c, thus c <= (-1+(1+4*(i+1))**0.5)/2
-                if (i-idxs[(curr-c)-1]-c)-c**2+1 >= 1:
-                    result += min(min(idxs[curr-c], i)-idxs[(curr-c)-1], (i-idxs[(curr-c)-1]-c)-c**2+1)
+                if c**2 <= (i-idxs[(curr-c)-1])-c:
+                    result += min(min(idxs[curr-c], i)-idxs[(curr-c)-1], ((i-idxs[(curr-c)-1])-c)-c**2+1)
         return result