Update minimum-number-of-flips-to-make-the-binary-string-alternating.cpp

Author: kamyu104

C++/minimum-number-of-flips-to-make-the-binary-string-alternating.cpp

@@ -6,7 +6,8 @@ class Solution {
     int minFlips(string s) {
         int result = numeric_limits<int>::max();
         int cnt1 = 0, cnt2 = 0;
-        for (int i = 0; i + 1 < 2 * size(s); ++i) {
+        int l = (size(s) % 2) ? 2 * size(s) - 1 : size(s);
+        for (int i = 0; i < l; ++i) {
             if (i >= size(s)) {
                 cnt1 -= (s[i % size(s)] - '0') ^ ((i - size(s)) % 2) ^ 0;
                 cnt2 -= (s[i % size(s)] - '0') ^ ((i - size(s)) % 2) ^ 1;