Codeforces- 1535C-Unstable String

Problem - C - Codeforces

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

int main(){
    int t;
	string s;
	int c[2];
	int len;
    cin >> t;
	cout << t << endl;
	for(int i = 0 ; i < t ; i++){
		cin >> s;
		len = s.size();
		long long cnt = 0;
		//c[0] -> '0'串：偶数位为0，奇数位为1. (0101010101...)
		//c[1] -> '1'串：奇数位为0，偶数位为1. (1010101010...)
		//c[x] -> 目标字符串中为x串子串的字符个数 (x = '0','1')
        //字符串要unstable则一定只包含'0'串和'1'串其中一个的子串，即c[0]和c[1]必有一个为0
		c[0] = c[1] = 0;
		if(s[0] != '?') c[(s[0]-'0')%2]++;
		int l = 0, r = 0;
		for(l = 0 ; l < len ; l++){ //每次循环l向后移一位
			while(r < len && (c[0] == 0 || c[1] == 0)){ //当前字符串的所有字符都仅为'0'串的子串/或仅为'1'串的子串
				r++; //当前字符串为unstable，将最后一位向后移
				
				if(r < len && s[r] != '?') c[(s[r]-'0' + r)%2]++;
				//s[r]为偶数位‘0’或奇数位‘1’则'0'串个数+1；s[r]为偶数位‘1’或奇数位‘0’则'1'串个数+1
			}
			
			//l到r-1为unstable，（l-1，r-1），（l-2，r-1），…，（r-1）都为unstable，cnt加r-1-l+1个
			cnt += r-l;
			//字符串中删除第l位，同时将对应'0'串/'1'串中的字符个数-1；
			if(s[l] != '?') c[(s[l]-'0' + l)%2]--;
		}
		cout << cnt << endl;
	}
}