回文自动机模板。
题意:求本质不同的回文子串个数。
分析:构建回文树,所求即回文树节点个数-2(除去两个长度为0的根节点)
各变量代表的意义在代码中已经注释
代码如下
#include<bits/stdc++.h>
#define lowbit(x) ((x)&(-(x)))
#define ll long long
#define INF 2147483647
#define CLR(a) memset(a, 0, sizeof(a))
using namespace std;
const int maxn=2e6+10;
struct pamtree{
int nxt[maxn][27],fail[maxn];//回文树与失配指针
int vis[maxn];//i表示的回文字符串在整个字符串中出现了多少次
int num[maxn];//i表示的回文字符串中有多少个本质不同的字符串(包括本身)
int len[maxn];//节点i的回文长度
int S[maxn];//存放添加的字符
int last;//最新添加的回文
int n;//字符数组指针
int p;//节点指针 (p-2表示有多少本质不同的回文子串)
int newnode(int l){
memset(nxt[p],0,sizeof(nxt[p]));
vis[p]=num[p]==0;
len[p]=l;
return p++;
}
void init(){
p=0;
newnode(0);newnode(-1);
last=0;
n=0;
S[n]=-1;
fail[0]=1;//指向第二棵树
}
int build(int x){//构建fail ,x表示最大回文(不包含自身)节点
while(S[n-len[x]-1]!=S[n]) x=fail[x];
return x;
}
void add(int c){
c-='a';//WA
S[++n]=c;
int cur=build(last);
if(!nxt[cur][c]){如果这个回文串没有出现过,说明出现了一个新的本质不同的回文串
int now=newnode(len[cur]+2);
fail[now]=nxt[build(fail[cur])][c];
nxt[cur][c]=now;
num[now]=num[fail[now]]+1;
}
last=nxt[cur][c];
vis[last]++;
}
void count(){
for(int i=p-1;i>=0;i--)
vis[fail[i]]+=vis[i];
}
}pam;
char s[maxn];
int main() {
int t;cin>>t;
for(int w=1;w<=t;w++){
pam.init();
scanf("%s",s);
int len=strlen(s);
for(int i=0;i<len;i++) pam.add(s[i]);
printf("Case #%d: %d\n",w,pam.p-2);
}
return 0;
}