2019徐州网络赛G.Colorful String (回文树+dfs)

题意：

每个回文串的价值为回文串出现的次数乘上回文串中不同字符的个数
求一个串的价值

分析:

不会回文树直接没了
会的话应该是简单题

建树从两个树根dfs就行了

code：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
const int maxm=3e5+5;
struct Pam{
    int nt[maxm][26];//下一个节点
    int fail[maxm];//失配指针
    int cnt[maxm];//节点i表示的本质不同的串的个数(最后cal完才是准确的)
    int num[maxm];//节点i表示的回文串的回文后缀个数
    int len[maxm];//节点i表示的回文串长度
    int s[maxm];//存放添加的字符
    int last;//新添加一个字母后所形成的最长回文串对应节点
    int n;//添加的字符数，也是s数组的长度
    int p;//节点指针，表示节点数量
    int newnode(int x){//创建新节点
        for(int i=0;i<26;i++){
            nt[p][i]=0;
        }
        cnt[p]=num[p]=0;
        len[p]=x;//长度为x
        return p++;
    }
    void init(){//初始化
        p=n=last=0;
        newnode(0);//偶数长度树
        newnode(-1);//奇数长度树
        s[n]=-1;//设置第一个字符为-1(也可以是其他不可能被匹配的字符)
        fail[0]=1;
    }
    int getfail(int x){//利用fail找到匹配的节点
        while(s[n-len[x]-1]!=s[n])x=fail[x];
        return x;
    }
    void add(int c){
        c-='a';
        s[++n]=c;
        int cur=getfail(last);
        if(!nt[cur][c]){//如果没有就新建一个节点
            int now=newnode(len[cur]+2);//从cur拓展而来
            fail[now]=nt[getfail(fail[cur])][c];
            nt[cur][c]=now;
            num[now]=num[fail[now]]+1;
        }
        last=nt[cur][c];
        cnt[last]++;
    }
    void cal(){
        for(int i=p-1;i>=0;i--){
            cnt[fail[i]]+=cnt[i];
        }
    }
}p;
char s[maxm];
int mark[26];
int sum=0;
ll dfs(int x){
    ll ans=0;
    for(int i=0;i<26;i++){
        int ntt=p.nt[x][i];
        if(ntt){
            mark[i]++;
            if(mark[i]==1){
                sum++;
            }
            ans+=sum*p.cnt[ntt]+dfs(ntt);
            mark[i]--;//回溯的时候删除
            if(mark[i]==0){
                sum--;
            }
        }
    }
    return ans;
}
int main(){
    scanf("%s",s);
    p.init();
    for(int i=0;s[i];i++){
        p.add(s[i]);
    }
    p.cal();
    ll ans=dfs(0)+dfs(1);
    printf("%lld",ans);//用%I64d会wa
    return 0;
}