C. Unstable String（双指针）

Posted 2021-06-24 zjj0624

题意
给你一个字符串s，定义如果一个字符串任意两个位置符号都不同，且只有0，1组成的字符串是不稳定的。
定义由0，1，？，组成的字符串，且？可以任意的转换成0和1，使字符串可以变成不稳定的字符串叫做美丽字符串。
给一个字符串求有多少子字符串可以形成美丽的字符串。
思路
假如有一个区间[l,r]，如果这个区间内的字符串是美丽的，那以r结尾的美丽子字符串的贡献就是r-l+1.
我们每次让右指针向右遍历，如果当前区间是美丽字符串，那以r结尾的子符串产生的贡献就是r-l+1，如果遍历的过程中当前的字符串不是美丽字符串，那我们就让左指针向右。
但是我们怎么判断当前的区间是不是美丽的呢？
我们判断一个字符串是不是美丽的，其实只需要判断当前区间是不是只有奇数位置上有一种字符，偶数位置上是另一种字符，？号的时候可以忽略，因为有?可以变成奇数和偶数，不会影响判断。
记得答案要开LL。
如果是全是？的串，最终的结果是n*（n+1）/2是会超出int的。
代码

#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        string s;
        cin>>s;
        int ji_1=0,ji_0=0,ou_1=0,ou_0=0;
        int l=0;
        ll ans=0;
        for(int i=0 ; i<s.length() ; i++)
        {
            if(s[i]=='0')
            {
                if(i%2==0) ou_0++;
                else ji_0++;
            }
            else if(s[i]=='1')
            {
                if(i%2==0) ou_1++;
                else ji_1++;
            }
            while((ou_0&&ji_0)||(ji_1&&ou_1)||(ji_1&&ji_0)||(ou_1&&ou_0))
            {
                if(s[l]=='0')
                {
                    if(l%2==0) ou_0--;
                    else ji_0--;
                }
                else if(s[l]=='1')
                {
                    if(l%2==0) ou_1--;
                    else ji_1--;
                }
                l++;
            }
           // cout<<l<<" "<<i<<endl;
            ans+=(1ll+i-l);
        }
        cout<<ans<<endl;
    }
    return 0;
}