携程笔试Java 0907 第四题详解

#include< iostream>
#include< string>
using namespace std;

int main() 
{
    string a;
    cin>>a;
    long long ans = 0;
    int t = 0;
    for(int i=0;i< a.size();i++)
    {
        if(a[i] == '0')
        {  
            if(t<0) t = 0;
            t++;
            if(t>0) ans+=t;
        }
        else if(a[i] == '1')
        {
            t--;
            if(t>0) ans+=t;
        }
    }
    cout<< ans << endl;
}