BJ模拟 最长公共前缀【树上后缀数组】

题目大意：

bzoj3879的树上版本，trie树上每个节点到根的路径都看做是一个后缀。

解题思路：

建一波树上后缀数组，再用bzoj3879的方法（这里）即可。
第一次写树上后缀数组，全靠yy，好多地方被普通后缀数组带偏，细节见代码。

#include<bits/stdc++.h>
#define ll unsigned long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=100005,base=31;
int n,m,q,fa[N][20],dep[N];
int tot,first[N],nxt[N<<1],to[N<<1],ch[N<<1];
int sa[N],sa2[N],rk[N],rk2[N],height[N],tp[N],c[N],mn[N<<1][20],Log[N];
int top,a[N],h[N],stk[N],L[N],R[N];
ll g[N][20],bin[1<<20];
inline bool cmp(const int &a,const int &b){return rk[a]<rk[b];}
inline void add(int x,int y,int z)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y,ch[tot]=z;
}
void dfs(int u)
{
    for(int i=1;i<20;i++)fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int i=1;i<20;i++)g[u][i]=g[u][i-1]*bin[1<<i-1]+g[fa[u][i-1]][i-1];
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(v==fa[u][0])continue;
        fa[v][0]=u,g[v][0]=ch[e],dep[v]=dep[u]+1;
        dfs(v);
    }
}
inline void Rsort(int *rk,int *sa)
{
    for(int i=0;i<=m;i++)c[i]=0;
    for(int i=1;i<=n;i++)c[rk[tp[i]]]++;
    for(int i=1;i<=m;i++)c[i]+=c[i-1];
    for(int i=n;i>=1;i--)sa[c[rk[tp[i]]]--]=tp[i];
}
inline int get_height(int x,int y)
{
    int res=0;
    for(int i=19;i>=0;i--)if(min(dep[x],dep[y])>=(1<<i))
    {
        if(g[x][i]==g[y][i])
            res+=(1<<i),x=fa[x][i],y=fa[y][i];
    }
    return res;
}
inline void build_SA()
{
    m=26;
    for(int i=1;i<=n;i++)rk[i]=g[i][0],tp[i]=i;
    Rsort(rk,sa);
    for(int k=1,w=0;k<n;k<<=1,w++)
    {
        int j=0;
        for(int i=1;i<=n;i++)rk2[i]=rk[fa[i][w]];
        for(int i=1;i<=n;i++)tp[i]=i;
        Rsort(rk2,sa2);
        for(int i=1;i<=n;i++)tp[i]=sa2[i];
        Rsort(rk,sa);swap(rk,tp);rk[sa[1]]=j=1;
        for(int i=2;i<=n;i++)
            rk[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[fa[sa[i]][w]]==tp[fa[sa[i-1]][w]]?j:++j);
        m=j;
    }
    for(int i=1;i<=n;i++)rk[sa[i]]=i;
    height[1]=0;
    for(int i=2;i<=n;i++)
        height[rk[i]]=get_height(i,sa[rk[i]-1]);
}
inline void build_ST()
{
    Log[1]=0;for(int i=2;i<=n;i++)Log[i]=Log[i>>1]+1;
    for(int i=1;i<=n;i++)mn[i][0]=height[i];
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i<=n;i++)
            mn[i][j]=min(mn[i][j-1],mn[i+(1<<j-1)][j-1]);
}
inline int lcp(int i,int j)
{
    if(i==j)return dep[i];
    int l=rk[i]+1,r=rk[j];
    int t=Log[r-l+1];
    return min(mn[l][t],mn[r-(1<<t)+1][t]);
}
int main()
{
    //freopen("lx.in","r",stdin);
    //freopen("lx.out","w",stdout);
    bin[0]=1;for(int i=1;i<=(1<<20);i++)bin[i]=bin[i-1]*base;
    n=getint();
    for(int i=1;i<n;i++)
    {
        int x=getint(),y=getint(),z=getchar()-'a'+1;
        add(x,y,z),add(y,x,z);
    }
    dfs(1);
    build_SA(),build_ST();
    q=getint();
    while(q--)
    {
        int k=getint(),top;ll ans=0;
        for(int i=1;i<=k;i++)a[i]=getint();
        sort(a+1,a+k+1,cmp);
        for(int i=1;i<k;i++)h[i]=lcp(a[i],a[i+1]);
        h[0]=h[k]=-1;
        stk[top=0]=0;
        for(int i=1;i<k;i++)
        {
            while(h[stk[top]]>h[i])top--;
            L[i]=stk[top],stk[++top]=i;
        }
        stk[top=0]=k;
        for(int i=k-1;i;i--)
        {
            while(h[stk[top]]>=h[i])top--;
            R[i]=stk[top],stk[++top]=i;
        }
        for(int i=1;i<k;i++)ans+=1ll*h[i]*(i-L[i])*(R[i]-i);
        printf("%lld\n",ans);
    }
    return 0;
}