Description
对于一个01字符串,如果将这个字符串0和1取反后,再将整个串反过来和原串一样,就称作“反对称”字符串。比如00001111和010101就是反对称的,1001就不是。
现在给出一个长度为N的01字符串,求它有多少个子串是反对称的。
Input
第一行一个正整数N (N <= 500,000)。第二行一个长度为N的01字符串。
题解:
考虑回文自动机,由于反对称字符串一定是偶数,所以我们只需要维护偶回文树即可(即如果跳到root=1就停止跳fail),对于匹配反对称字符串,只需要修改一下回文自动机的匹配模式,让左右不相等,最后统计一下出现次数就好了。
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 5e5+50;
inline int read(){
int x=0,f=1; char ch=getchar();
while(ch<'0' || ch>'9'){ if(ch=='-') f=-1; ch = getchar(); }
while(ch>='0' && ch<='9') x=(x<<1)+(x<<3)+ch-48,ch=getchar();
return x*f;
}
char s[MAXN];
int nxt[MAXN][2],fail[MAXN],len[MAXN];
int last,tot,cnt[MAXN],a[MAXN];
inline void Init(){
last=tot=1;
len[1]=-1,len[0]=0;
fail[1]=0,fail[0]=1;
}
inline bool check(int en,int root){
if((a[en]==a[en-len[root]-1] || en-len[root]-1==0) && root!=1) return true;
return false;
}
inline void Insert(int x,int en){
int root = last;
while(check(en,root)) root=fail[root];
if(root==1) { last=0;return; }
if(!nxt[root][x]){
int sz = ++tot;
len[sz] = len[root]+2;
int tmp = fail[root];
while(check(en,tmp)) tmp=fail[tmp];
if(tmp==1) fail[sz]=0;
else fail[sz] = nxt[tmp][x];
nxt[root][x] = sz;
}
last = nxt[root][x];
++cnt[last];
}
inline void Solve(){
int ans = 0;
for(int i=tot;i;i--) cnt[fail[i]] += cnt[i],ans += cnt[i];
printf("%d\n",ans);
}
int main(){
int n = read();
scanf("%s",s+1);
Init();
for(int i=1;i<=n;i++) a[i]=s[i]-'0';
for(int i=1;i<=n;i++) Insert(a[i],i);
Solve();
return 0;
}