携程第四题，不清楚是不是最简洁

int main() {
    string s;
    cin>>s;
    long ans=0;
    int t=0;
    for (char c : s) {
        if (c=='0') {
            t++;
            ans+=t;
        } else {
            t--;
            if (t<0) t=0;
            ans+=t;
        }
    }
    cout<<ans<<endl;
}