Submission #68340080

---

Source Code Expand

#include<bits/stdc++.h>
using ll=long long;
using namespace std;
int read(){int f=0,x=0;char a=getchar();while(!isdigit(a)){if(a=='-')f=1;a=getchar();}while(isdigit(a)){x=x*10+a-'0';a=getchar();}if(f)return -x;return x;}
int main()
{
    //freopen(in.in,r,stdin);
    //freopen(out.out,w,stdout);
    ios::sync_with_stdio(false);
    cin.tie(0);
    int tttt=1;
    //cin>>tttt;
    while(tttt--)
    {
        ll n;cin>>n;
        string s;cin>>s;
        ll jo=0,jj=0,oo=0,oj=0,ans=0;
        for(int i=0;i<n;i++)
        {
            if(s[i]=='1')
            {
                swap(jo,oj);
                swap(jj,oo);
                jj++;
                ans+=jj+oo;
            }
            else
            {
                swap(jo,oo);
                swap(jj,oj);
                jo++;
                ans+=jj+oo;
            }
            //cout<<jo<<' '<<jj<<' '<<oo<<' '<<oj<<'\n';
        }
        cout<<ans<<'\n';
    }
    return 0;
}

Submission Info

Judge Result