[Poi2010]Antisymmetry（回文自动机）

2084: [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

解：

感觉hash做简单些，但是既然在学回文自动机，我们就写一写回文自动机。
这是一道很好的回文自动机变形的题目。首先我们把限制条件改为两个不相等才放进回文自动机。同时在回文自动机中我们只保存长度为偶数的串。这一点如何实现？
由于我们每次增量是长度加2，所以只要自动机中不记录长度为1的回文串即可。我们不建长度为1的回文串，当长度为1的回文串出现时，只需把last指向0号节点即可。

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 19930726
using namespace std;
struct lxy{
    int to[2],num,fail,len;
    bool operator < (const lxy &QAQ)const{
      return len<QAQ.len;
    } 
}b[1000005];

char s[1000005];
int root1=1,root2=0,len,cnt=1,last=1;
long long ans,q;

void insert(){
    int p;
    b[root2].fail=root1;
    b[root1].len=-1;
    for(int i=1;i<=len;i++){
        for(p=last;(s[i]==s[i-b[p].len-1]||i-b[p].len==1)&&p!=1;p=b[p].fail);
        if(p==1){last=0;continue;}
        if(b[p].to[s[i]-'0']==0){
            b[++cnt].len=b[p].len+2;
            int k;
            for(k=b[p].fail;(s[i]==s[i-b[k].len-1]||i-b[k].len==1)&&k!=1;k=b[k].fail);
            if(k==1) b[cnt].fail=0;else b[cnt].fail=b[k].to[s[i]-'0'];
            b[p].to[s[i]-'0']=cnt;
        }
        b[b[p].to[s[i]-'0']].num++;
        last=b[p].to[s[i]-'0'];
    }
}

long long qui(long long x,long long a){
    long long ret=1;
    while(a!=0){
        if((a&1)==1) ret=ret*x%mod;
        x=x*x%mod;
        a=a>>1; 
    }
    return ret;
}

int main()
{
    scanf("%d",&len);
    scanf("%s",s+1);s[0]='#';
    len=strlen(s+1);
    insert();
    for(int i=cnt;i>=2;i--){
        b[b[i].fail].num+=b[i].num;
        ans+=b[i].num;
    }
    printf("%lld",ans);
}