CF1428F-Fruit Sequences

题目链接https://codeforces.com/contest/1428/problem/F
参考题解https://www.cnblogs.com/zkyJuruo/p/13833960.html
OI爷，永远滴神！
思路
对于01字符串从右向左遍历，如果当前$s[i]=0$，那么对序列没有任何贡献，如果$s[i]=1$，如果出现连续的1，设在第i位该值为$h[i]$，那么从当前位到上一次出现$h[i]$之间的值，需要全部加上1 。例举一个样例就是110011，如果此时$i=2$，那么从$[2,4]$这一段范围，它的结果就会+3，而$[5,6]$的代价还是不变的。如果$i=1$，那么从$[1,5]$的范围之内代价都会+1，结果就是+5。由此看来就只要看当前出现的$h[i]$的这个值，在上一次所出现的位置在哪里，那么代价就会增加$h[i]-i$，如果不存在这样的位置，那么代价就相当于直接加上$n-i+1$。
开一个pos数组记录每一个h[i]最后出现的位置。因为在连续求1的过程中，其最后的位置恰好是反一反的，例如110011，h[5]=2,h[6]=1,但是在遍历到位置2的时候，h[2]=1所对应1应该是第5位，h[1]=2所对应的位置应该是第6位，所以在发现s[i]=0 && s[i+1]=1的时候，要把pos数组的一部分reverse一下。
代码

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

typedef long long LL;
#define int LL
typedef pair<int, int> PII;
const int inf = 0x3f3f3f3f;
const int N = 5e5 + 10;
const int mod = 1e9 + 7;

char s[N];
int pos[N], h[N];

void solve() {
    int n;
    scanf("%lld%s", &n, s + 1);
    int res = 0, sum = 0;

    for(int i = n; i; i--) {
        h[i] = s[i] == '0' ? 0 : h[i + 1] + 1;
        if(s[i] == '1') {
            if(!pos[h[i]]) sum += n - i + 1;
            else sum += pos[h[i]] - i;
            pos[h[i]] = i;
        }
        else if(s[i] == '0' && s[i + 1] == '1') {
        	int x = h[i + 1];
        	reverse(pos + 1, pos + 1 + x);
        }
        res += sum;
        // printf("i = %d sum = %d\n", i, sum);
    }
    printf("%lld\n", res);
}

signed main() {
    // freopen("in.txt", "r", stdin);
    // int t; cin >> t; while(t--)
    solve();
    return 0;
}