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>='0' && c<='9';}
int qread()
{
    char c;int s=0,t=1;while (!isdigit(c=getchar())) (c=='-' && (t=-1));
    do s=s*10+c-'0'; while (isdigit(c=getchar())); return s*t;
}

int n;
#define maxn 300011
struct Edge{int to,next;}edge[maxn<<1];int first[maxn],le=2;
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<<1][26],size,val[maxn<<1];
    Trie() {memset(ch[0],0,sizeof(ch[0]));size=0;}
    int id(char c) {return c-'a';}
    void up(int x)
    {
        val[x]=1;
        for (int i=0;i<26;i++)
            if (ch[x][i]) val[x]+=val[ch[x][i]];
    }
    int New(char c)
    {
        size++;memset(ch[size],0,sizeof(ch[size]));
        val[size]=2;
        size++;memset(ch[size],0,sizeof(ch[size]));
        ch[size-1][id(c)]=size;
        val[size]=1;
        return size-1;
    }
    int combine(int x,int y)
    {
        if (!x || !y) return x+y;
        for (int i=0;i<26;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=0;
    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=0;i<26;i++) t.ch[u][i]=t.ch[root[base]][i];
        t.up(u);t.up(root[x]);
    }
    dif[x]=t.val[root[x]]-1;
//    cout<<x<<' '<<dif[x]<<endl;
}

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