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);
}