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

Author: kamyu104

C++/count-the-number-of-substrings-with-dominant-ones.cpp

@@ -19,7 +19,7 @@ class Solution {
             }
             const int l = (-1 + sqrt(1 + 4 * (i + 1))) / 2;  // since c^2 <= (i+1)-c, thus c <= (-1+(1+4*(i+1))**0.5)/2
             for (int c = 0; c <= min(l, curr - 1); ++c) {
-                if (((i - idxs[(curr - c) - 1]) - c) - c * c + 1 >= 1) {
+                if (c * c <= (i - idxs[(curr - c) - 1]) - c) {
                     result += min(min(idxs[curr - c], i) - idxs[(curr - c) - 1], ((i - idxs[(curr - c) - 1]) - c) - c * c + 1);
                 }
             }