Bit string

input/code.cpp: In function 'int main()':
input/code.cpp:40:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0; i < s.size(); i++){
                    ~~^~~~~~~~~~

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define INF 9223372036854775807

string s;

int memo[2][500009];

int dp(int parity, int index){
    if(index < 0) return 0;
    if(memo[parity][index] != -1) return memo[parity][index];
    if(parity == 0){
        if(s[index] == '1'){
            return memo[parity][index] = dp(1,index-1);
        }else{
            return memo[parity][index] = dp(0,index-1)+1;
        }
    }else{
        if(s[index] == '0'){
            return memo[parity][index] = dp(1,index-1);
        }else{
            return memo[parity][index] = dp(0,index-1)+1;
        }
    }

}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> s;
    for(int i = 0; i < 2; i++){
        for(int j = 0; j < 500009; j++){
            memo[i][j] = -1;
        }
    }
    int tot = 0;
    for(int i = 0; i < s.size(); i++){
        //cout << i << ' ' << dp(0,i) << endl;
        tot+=dp(0,i);
    }
    cout << tot << endl;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED

Test 14

Group: 2

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: ACCEPTED

Test 17

Group: 3

Verdict: ACCEPTED

Test 18

Group: 3

Verdict: ACCEPTED

Test 19

Group: 3

Verdict: ACCEPTED

Test 20

Group: 3

Verdict: ACCEPTED

Test 21

Group: 3

Verdict: ACCEPTED