bzoj2084: [Poi2010]Antisymmetry 回文自动机

bzoj2084: [Poi2010]Antisymmetry

Description

对于一个01字符串，如果将这个字符串0和1取反后，再将整个串反过来和原串一样，就称作“反对称”字符串。比如00001111和010101就是反对称的，1001就不是。
现在给出一个长度为N的01字符串，求它有多少个子串是反对称的。

Input

第一行一个正整数N (N <= 500,000)。第二行一个长度为N的01字符串。

Output

一个正整数，表示反对称子串的个数。

Sample Input

8
11001011

Sample Output

7
hint
7个反对称子串分别是：01(出现两次), 10(出现两次), 0101, 1100和001011

分析

题解看这里：bzoj 2084: Antisymmetry 回文自动机

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<cmath>
using namespace std;
const int N = 550000;
int s[N], ch[N][2], f[N], len[N], siz[N], l, last, sz, ans;
int F(int p) {for(; p != 1 && (s[l - len[p] - 1] == s[l] || !(l - len[p] - 1)); p = f[p]); return p;}
void Extend(int c) {
    s[++l] = c; int p = F(last), np;
    if(p == 1) {last = 0; return;}
    if(!ch[p][c]) {
        len[np = ++sz] = len[p] + 2;
        f[np] = ch[F(f[p])][c]; ch[p][c] = np;
    }
    ++siz[last = ch[p][c]];
}

int main() {
    int n; scanf("%d", &n); f[0] = 1; sz = 1; len[1] = -1;
    for(int i = 1;i <= n; ++i) {
        char ch; for(ch = getchar(); ch != '0' && ch != '1'; ch = getchar()) ;
        Extend(ch - '0');
    }
    for(int i = sz; i; --i) siz[f[i]] += siz[i];
    for(int i = 2;i <= sz; ++i) if(~len[i] & 1) ans += siz[i];
    printf("%d\n", ans);
    return 0;
}