Palindromeness CodeChef - PALPROB

传送门

分析

有中文题面所以就不写题目大意了

我们先建出回文树

然后根据fail信息建出一棵树

从根向下dfs，中间记录每一个len出现在哪个节点即可

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
int n,fail[],son[][],sum[],len[],cnt,last;
int val[],pos[],nxt[],head[],to[],tot;
long long Ans;
char s[];
inline void add(int x,int y){nxt[++tot]=head[x];head[x]=tot;to[tot]=y;}
inline void new_node(int x){len[++cnt]=x;sum[cnt]=;}
inline int get_fail(int x,int n){while(s[n]!=s[n-len[x]-])x=fail[x];return x;}
inline void build(){
    for(int i=;i<=n;i++){
      int x=s[i]-'a',now=get_fail(last,i);
      if(!son[now][x]){
          new_node(len[now]+);
          fail[cnt]=son[get_fail(fail[now],i)][x];
          son[now][x]=cnt;
      }
      last=son[now][x];
      sum[last]++;
    }
}
inline void dfs(int x){
    if(x>)pos[len[x]]=x,Ans+=(long long)(val[x]=val[pos[len[x]/]]+)*sum[x];
    for(int i=head[x];i;i=nxt[i])dfs(to[i]);
    if(x>)pos[len[x]]=;
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
      scanf("%s",s+);
      n=strlen(s+);
      memset(sum,,sizeof(sum));
      memset(son,,sizeof(son));
      memset(fail,,sizeof(fail));
      memset(len,,sizeof(len));
      memset(head,,sizeof(head));
      tot=;
      Ans=;
      cnt=-;
      new_node(),new_node(-);
      fail[]=fail[]=;
      last=;
      build();
      add(,);
      for(int i=cnt;i>;i--)sum[fail[i]]+=sum[i],add(fail[i],i);
      dfs();
      printf("%lld\n",Ans);
    }
    return ;
}