來自FallDream的博客，未經(jīng)允許， 請勿轉(zhuǎn)載，謝謝。

---

幽香是全幻想鄉(xiāng)里最受人歡迎的萌妹子，這天，是幽香的2600歲生日，無數(shù)幽香的粉絲到了幽香家門前的太陽花田上來為幽香慶祝生日。?

粉絲們非常熱情，自發(fā)組織表演了一系列節(jié)目給幽香看。幽香當(dāng)然也非常高興啦。?

這時幽香發(fā)現(xiàn)了一件非常有趣的事情，太陽花田有n塊空地。在過去，幽香為了方便，在這n塊空地之間修建了n-1條邊將它們連通起來。也就是說，這n塊空地形成了一個樹的結(jié)構(gòu)。?

有n個粉絲們來到了太陽花田上。為了表達(dá)對幽香生日的祝賀，他們選擇了c中顏色的衣服，每種顏色恰好可以用一個0到c-1之間的整數(shù)來表示。并且每個人都站在一個空地上，每個空地上也只有一個人。這樣整個太陽花田就花花綠綠了。幽香看到了，感覺也非常開心。?

粉絲們策劃的一個節(jié)目是這樣的，選中兩個粉絲A和B（A和B可以相同），然后A所在的空地到B所在的空地的路徑上的粉絲依次跳起來（包括端點(diǎn)），幽香就能看到一個長度為A到B之間路徑上的所有粉絲的數(shù)目（包括A和B）的顏色序列。一開始大家打算讓人一兩個粉絲（注意：A,B和B,A是不同的，他們形成的序列剛好相反，比如紅綠藍(lán)和藍(lán)綠紅）都來一次，但是有人指出這樣可能會出現(xiàn)一些一模一樣的顏色序列，會導(dǎo)致審美疲勞。?

于是他們想要問題，在這個樹上，一共有多少可能的不同的顏色序列（子串）幽香可以看到呢？?

太陽花田的結(jié)構(gòu)比較特殊，只與一個空地相鄰的空地?cái)?shù)量不超過20個。?

n<=100000 c<=10

因?yàn)槿~子結(jié)點(diǎn)不超過20個，所以可以從每個葉子結(jié)點(diǎn)開始dfs，把節(jié)點(diǎn)插入同一個后綴自動機(jī)上。最后在后綴自動機(jī)上統(tǒng)計(jì)子串?dāng)?shù)量即可。

復(fù)雜度O(10*20*n)

#include<iostream> #include<cstdio> #include<algorithm> #define MN 100000 #define ME 4000000 #define ll long long using namespace std; inline int read() {int x = 0 , f = 1; char ch = getchar();while(ch < '0' || ch > '9'){ if(ch == '-') f = -1; ch = getchar();}while(ch >= '0' && ch <= '9'){x = x * 10 + ch - '0';ch = getchar();}return x * f; }ll sum[ME+5]; int c[ME+5][11],n,m,step[ME+5],fail[ME+5],v[MN+5],sa[ME+5],in[MN+5],col[MN+5],cnt=0,head[MN+5]; struct edge{int to,next;}e[MN*2+5]; inline void ins(int f,int t) {++in[f];++in[t];e[++cnt]=(edge){t,head[f]};head[f]=cnt;e[++cnt]=(edge){f,head[t]};head[t]=cnt; }int Ins(int p,int x) {int np=++cnt;step[np]=step[p]+1;for(;p&&!c[p][x];p=fail[p]) c[p][x]=np;if(!p) fail[np]=1;else{int q=c[p][x];if(step[q]==step[p]+1) fail[np]=q;else{int nq=++cnt;step[nq]=step[p]+1;fail[nq]=fail[q];fail[q]=fail[np]=nq;for(int i=0;i<m;++i) c[nq][i]=c[q][i];for(;c[p][x]==q;p=fail[p]) c[p][x]=nq;}}return np; }void Solve(int x,int fa,int p) {int np=Ins(p,col[x]);for(int i=head[x];i;i=e[i].next)if(e[i].to!=fa) Solve(e[i].to,x,np); }int main() {n=read();m=read();for(int i=1;i<=n;++i) col[i]=read();for(int i=1;i<n;++i) ins(read(),read());cnt=1;for(int i=1;i<=n;++i) if(in[i]==1) Solve(i,0,1);for(int i=1;i<=cnt;++i) ++v[step[i]];for(int i=1;i<=MN;++i) v[i]+=v[i-1];for(int i=1;i<=cnt;++i) sa[v[step[i]]--]=i;for(int i=cnt;i;--i){sum[sa[i]]=sa[i]!=1;for(int j=0;j<m;++j) if(c[sa[i]][j])sum[sa[i]]+=sum[c[sa[i]][j]];}printf("%lld\n",sum[1]);return 0; }

轉(zhuǎn)載于:https://www.cnblogs.com/FallDream/p/bzoj3926.html

總結(jié)

以上是生活随笔為你收集整理的[bzoj3926][Zjoi2015]诸神眷顾的幻想乡的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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