題目鏈接：點(diǎn)擊查看

題目大意：給出一個(gè)字典樹，現(xiàn)在需要求出字典樹上所有的回文串做出的貢獻(xiàn)，為 出現(xiàn)次數(shù)*回文串長(zhǎng)度，求出這個(gè)答案

題目鏈接：可以直接在字典樹上dfs然后維護(hù)貢獻(xiàn)，不過這就涉及到了回文自動(dòng)機(jī)的刪除功能，其實(shí)實(shí)現(xiàn)起來很簡(jiǎn)單，只需要將last變成數(shù)組記錄深度即可，每個(gè)節(jié)點(diǎn)的貢獻(xiàn)為其fail所指的結(jié)點(diǎn)的貢獻(xiàn)加上其本身的長(zhǎng)度，因?yàn)槭亲铋L(zhǎng)回文后綴，所以肯定包含了fail所指的結(jié)點(diǎn)的貢獻(xiàn)，再加上自己本身的長(zhǎng)度就是每個(gè)節(jié)點(diǎn)的貢獻(xiàn)了

代碼：

#pragma comment(linker, "/STACK:102400000,102400000") #include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<queue> #include<map> #include<set> #include<sstream> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=2e6+100; //字符串長(zhǎng)度最大值struct Edge {int to,next; }edge[N];int head[N],cnt;LL ANS;void addedge(int u,int v) {edge[cnt].to=v;edge[cnt].next=head[u];head[u]=cnt++; }char s[N];struct Palindrome_tree {int nxt[N][26];int fail[N]; // 當(dāng)前節(jié)點(diǎn)最長(zhǎng)回文后綴的節(jié)點(diǎn)int len[N]; // 當(dāng)前節(jié)點(diǎn)表示的回文串的長(zhǎng)度int cnt[N]; // 當(dāng)前節(jié)點(diǎn)回文串的個(gè)數(shù)， 在getcnt后可得到全部int sed[N]; // 以當(dāng)前節(jié)點(diǎn)為后綴的回文串的個(gè)數(shù)（并不是表示第i結(jié)尾的回文串的種類數(shù)，如果要求每個(gè)點(diǎn)結(jié)尾的數(shù)的回文串個(gè)數(shù)，得用last）int record[N]; //record記錄了節(jié)點(diǎn)回文串的結(jié)束位置char s[N];int tot; // 節(jié)點(diǎn)個(gè)數(shù)int last[N]; // 上一個(gè)節(jié)點(diǎn)LL ans[N];int n;//當(dāng)前字符串的長(zhǎng)度 void init(){tot = n = 0;memset(fail, 0, sizeof fail);memset(cnt, 0, sizeof cnt);memset(sed, 0, sizeof sed);memset(len, 0, sizeof len);memset(nxt, 0, sizeof nxt);memset(ans,0,sizeof(ans));}void build(){len[0] = 0, len[1] = -1; // 0為偶數(shù)長(zhǎng)度根， 1為奇數(shù)長(zhǎng)度根tot = 1, last[0] = 0;fail[0] = 1;}int getfail(int x, int n){while (s[n - len[x] - 1] != s[n]||n-len[x]-1<0) // 比較x節(jié)點(diǎn)回文串新建兩端是否相等//n-len[x]-1<0這個(gè)是我自己加的，多組的時(shí)候光第一個(gè)條件是不夠的，所以有錯(cuò)請(qǐng)手動(dòng)刪除x = fail[x]; // 若不同， 再比較x后綴回文串兩端return x;}void insert(char ch,int deep){int c = ch - 'a';//全小寫要用a 全大寫要用A 不然會(huì)錯(cuò)s[n=deep]=ch;int p = getfail(last[n-1], n);// 得到第i個(gè)字符可以加到哪個(gè)節(jié)點(diǎn)的兩端形成回文串if (!nxt[p][c]){tot++;len[tot] = len[p] + 2; // 在p節(jié)點(diǎn)兩端添加兩個(gè)字符fail[tot] = nxt[getfail(fail[p], n)][c]; //tot點(diǎn)的后綴回文，可以由上一個(gè)節(jié)點(diǎn)的后綴回文嘗試得到sed[tot] = sed[fail[tot]] + 1; // 以當(dāng)前節(jié)點(diǎn)為結(jié)尾的回文串個(gè)數(shù)nxt[p][c] = tot; // 新建節(jié)點(diǎn)ans[tot]=ans[fail[tot]]+len[tot];}last[n] = nxt[p][c]; // 當(dāng)前節(jié)點(diǎn)成為上一個(gè)節(jié)點(diǎn)cnt[last[n]]++; //當(dāng)前節(jié)點(diǎn)回文串++record[last[n]] = n;ANS+=ans[last[n]];}void get_cnt(){for (int i = tot; i > 0; i--)cnt[fail[i]] += cnt[i];//fail[i] 的節(jié)點(diǎn) 為 i 節(jié)點(diǎn)的后綴回文串， 所以個(gè)數(shù)相加} }tree;void dfs(int u,int deep) {for(int i=head[u];i!=-1;i=edge[i].next){int to=edge[i].to;tree.insert(s[to],deep);dfs(to,deep+1);} }void init() {tree.init();tree.build();memset(head,-1,sizeof(head));cnt=0;ANS=0; }int main() { // freopen("input.txt","r",stdin); // ios::sync_with_stdio(false);int w;cin>>w;while(w--){init();int n;scanf("%d",&n);for(int i=1;i<=n;i++){char ch[5];int fa;scanf("%s%d",ch,&fa);s[i]=ch[0];addedge(fa,i);}dfs(0,1);printf("%lld\n",ANS);}return 0; }

?

總結(jié)

以上是生活随笔為你收集整理的HDU - 5394 Trie in Tina Town(回文自动机+字典树)的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。