CC TSUBSTR：Substrings on a Tree——题解
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了CC TSUBSTR：Substrings on a Tree——题解相关的知识，希望对你有一定的参考价值。
https://www.codechef.com/problems/TSUBSTR
https://vjudge.net/problem/CodeChef-TSUBSTR
给一棵点权为字母的树，你只能从任意节点往下走得到一个字符串，求出可得到的不重复字符串数量和其中字典序第k小的字符串（在重新定义字母的字典序前提下）？
垃圾谷歌翻译坑我不浅。
广义后缀自动机处理之后就是BZOJ3998：[TJOI2015]弦论t=0的解法了。
注意空字符串也算。
#include<cstdio> #include<iostream> #include<queue> #include<cstring> #include<algorithm> #include<cctype> using namespace std; typedef long long ll; const int N=2e6+5; struct tree{ int a[26],fa,l; }tr[N]; struct node{ int to,nxt; }e[N]; char s[N]; int tot,cnt,head[N],pos[N]; ll a[N],w[N],size[N],sum[N]; bool vis[N]; queue<int>q; inline void add(int u,int v){ e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt; } inline int insert(int p,int c){ int np=tr[p].a[c]; if(np&&tr[np].l==tr[p].l+1)return np; np=++tot;tr[np].l=tr[p].l+1; for(;p&&!tr[p].a[c];p=tr[p].fa)tr[p].a[c]=np; if(!p)tr[np].fa=1; else{ int q=tr[p].a[c]; if(tr[p].l+1==tr[q].l)tr[np].fa=q; else{ int nq=++tot;tr[nq].l=tr[p].l+1; memcpy(tr[nq].a,tr[q].a,sizeof(tr[q].a)); tr[nq].fa=tr[q].fa;tr[q].fa=tr[np].fa=nq; for(;p&&tr[p].a[c]==q;p=tr[p].fa)tr[p].a[c]=nq; } } return np; } void bfs(int st){ tot=1; q.push(st);vis[st]=1; pos[st]=insert(1,s[st]-\'a\'); while(!q.empty()){ int u=q.front();q.pop(); for(int i=head[u];i;i=e[i].nxt){ int v=e[i].to; if(vis[v])continue; q.push(v);vis[v]=1; pos[v]=insert(pos[u],s[v]-\'a\'); } } } int main(){ int len,q; scanf("%d%d",&len,&q); scanf("%s",s+1); for(int i=1;i<len;i++){ int u,v; scanf("%d%d",&u,&v); add(u,v);add(v,u); } bfs(1); for(int i=1;i<=tot;i++)w[tr[i].l]++; for(int i=1;i<=len;i++)w[i]+=w[i-1]; for(int i=1;i<=tot;i++)a[w[tr[i].l]--]=i; for(int i=tot;i>=1;i--){ size[a[i]]=1; } for(int i=tot;i>=1;i--){ sum[a[i]]=size[a[i]]; for(int j=0;j<26;j++) if(tr[a[i]].a[j])sum[a[i]]+=sum[tr[a[i]].a[j]]; } printf("%lld\\n",sum[1]); for(int i=1;i<=q;i++){ ll k; scanf("%s%lld",s,&k); if(k>sum[1]){puts("-1");continue;} int now=1; while(k>size[now]){ k-=size[now]; for(int j=0;j<26;j++){ if(k>sum[tr[now].a[s[j]-\'a\']]){ k-=sum[tr[now].a[s[j]-\'a\']]; }else{ now=tr[now].a[s[j]-\'a\']; putchar(s[j]); break; } } } puts(""); } return 0; }
+++++++++++++++++++++++++++++++++++++++++++
+本文作者：luyouqi233。 +
+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+
+++++++++++++++++++++++++++++++++++++++++++
以上是关于CC TSUBSTR：Substrings on a Tree——题解的主要内容，如果未能解决你的问题，请参考以下文章
HDU 4455 Substrings（线性dp，很有意思）
Get a specific pixel coordinates where your mouse on （cc)