CF601D：Acyclic Organic Compounds

给n<=300000的树，每个点上有一个字母，一个点的权值为：从该点出发向下走到任意节点停下形成的不同字符串的数量，问最大权值。

题目本身还有一些奇怪要求在此忽略。。

Trie合并的模板题。

 #include<stdio.h>
 #include<string.h>
 #include<stdlib.h>
 #include<math.h>
 //#include<queue>
 #include<algorithm>
 #include<iostream>
 using namespace std;

 bool isdigit(char c) {return c>='' && c<='';}
 int qread()
 {
     char c;int s=,t=;while (!isdigit(c=getchar())) (c=='-' && (t=-));
     do s=s*+c-''; while (isdigit(c=getchar())); return s*t;
 }

 int n;
 #define maxn 300011
 struct Edge{int to,next;}edge[maxn<<];int first[maxn],le=;
 void in(int x,int y) {Edge &e=edge[le];e.to=y;e.next=first[x];first[x]=le++;}
 void insert(int x,int y) {in(x,y);in(y,x);}

 int c[maxn];
 char s[maxn];
 struct Trie
 {
     int ch[maxn<<][],size,val[maxn<<];
     Trie() {memset(ch[],,sizeof(ch[]));size=;}
     int id(char c) {return c-'a';}
     void up(int x)
     {
         val[x]=;
         for (int i=;i<;i++)
             if (ch[x][i]) val[x]+=val[ch[x][i]];
     }
     int New(char c)
     {
         size++;memset(ch[size],,sizeof(ch[size]));
         val[size]=;
         size++;memset(ch[size],,sizeof(ch[size]));
         ch[size-][id(c)]=size;
         val[size]=;
         return size-;
     }
     int combine(int x,int y)
     {
         if (!x || !y) return x+y;
         for (int i=;i<;i++)
             ch[x][i]=combine(ch[x][i],ch[y][i]);
         up(x);
         return x;
     }
 }t;

 int root[maxn],dif[maxn];
 inline void dfs(int x,int fa)
 {
     root[x]=t.New(s[x]);
     int base=;
     for (int i=first[x];i;i=edge[i].next)
     {
         const Edge &e=edge[i]; if (e.to==fa) continue;
         dfs(e.to,x);
         if (!base) base=e.to;
         else root[base]=t.combine(root[base],root[e.to]);
     }
     if (base)
     {
         int u=t.ch[root[x]][t.id(s[x])];
         for (int i=;i<;i++) t.ch[u][i]=t.ch[root[base]][i];
         t.up(u);t.up(root[x]);
     }
     dif[x]=t.val[root[x]]-;
 //    cout<<x<<' '<<dif[x]<<endl;
 }

 int main()
 {
     n=qread();
     for (int i=;i<=n;i++) c[i]=qread();
     scanf("%s",s+);
     for (int i=,x,y;i<n;i++)
     {
         x=qread(),y=qread();
         insert(x,y);
     }
     dfs(,);
     int ans=,cnt=;
     for (int i=;i<=n;i++)
     {
         if (c[i]+dif[i]>ans) ans=c[i]+dif[i],cnt=;
         else if (c[i]+dif[i]==ans) cnt++;
     }
     printf("%d\n%d\n",ans,cnt);
     return ;
 }