bzoj 3439 Kpm的MC密码（Trie+dfs序+主席树）

【题目链接】

http://www.lydsy.com/JudgeOnline/problem.php?id=3439

【题意】

给定若干串，问一个串的作为其后缀的给定串集合中的第k小。

【思路】

如果将每个串反向，则问题为一个串作为其前缀的给定串集合的第k小。

如果用Trie组织数据，则发现该串对应的集合即为以Trie上以尾节点为根的子树。我们按照dfs序构造序列，则问题即为连续区间上的第k大，可以用主席树求出。

注意有重复的串。

【代码】

 #include<set>
 #include<cmath>
 #include<queue>
 #include<vector>
 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 using namespace std;

 typedef long long ll;
 const int N = 1e6+;

 ll read() {
     char c=getchar();
     ll f=,x=;
     while(!isdigit(c)) {
         if(c=='-') f=-; c=getchar();
     }
     while(isdigit(c))
         x=x*+c-'',c=getchar();
     return x*f;
 }

 char s[N];
 int n,m,dfsc,L[N],R[N],dfs_list[N],sz;

 struct Trie {
     int ch[N][],sz;
     vector<int> num[N];
     Trie() {
         sz=;
         memset(ch,,sizeof(ch));
     }
     void insert(char *s,int val) {
         int u=;
         for(int i=;s[i];i++) {
             int c=s[i]-'a';
             if(!ch[u][c]) ch[u][c]=++sz;
             u=ch[u][c];
         }
         num[u].push_back(val);
     }
     void dfs(int u)
     {
         int tmp=dfsc+;
         FOR(i,,(int)num[u].size()-) {
             dfs_list[++dfsc]=num[u][i];
             L[num[u][i]]=tmp;
         }
         FOR(c,,) if(ch[u][c])
             dfs(ch[u][c]);
         FOR(i,,(int)num[u].size()-)
             R[num[u][i]]=dfsc;
     }
 } trie;

 struct Tnode {
     Tnode *ch[];
     int num;
     void * operator
         new(size_t size,Tnode *l,Tnode *r,int x) ;
 }*T[N],*mempool,*G;
 //´óСΪ 1<<15 µÄÄÚ´æ³Ø
 void *Tnode::operator new (size_t size,Tnode *l,Tnode*r,int x) {
     if(G==mempool) {
         G=new Tnode[<<];
         mempool=G+(<<);
         memset(G,,sizeof(Tnode)*(<<));
     }
     G->ch[]=l,G->ch[]=r,G->num=x;
     return G++;
 }

 Tnode* build(Tnode* p,int l,int r,int val) {
     int mid=l+r>>;
     if(l==r) return new (0x0,0x0,p->num+)Tnode;
     if(val<=mid)
         return new (build(p->ch[],l,mid,val),p->ch[],p->num+)Tnode;
     else
         return new (p->ch[],build(p->ch[],mid+,r,val),p->num+)Tnode;
 }
 int query(Tnode *x,Tnode *y,int l,int r,int rk) {
     int mid=l+r>>;
     if(l==r) return mid;
     int sum=y->ch[]->num - x->ch[]->num;
     if(rk<=sum) return query(x->ch[],y->ch[],l,mid,rk);
     else return query(x->ch[],y->ch[],mid+,r,rk-sum);
 }

 int main()
 {
     n=read();
     FOR(i,,n) {
         scanf("%s",s);
         int n=strlen(s);
         for(int j=;j*<n;j++)
             swap(s[j],s[n--j]);
         trie.insert(s,i);
     }
     trie.dfs();

     T[]=new (0x0,0x0,) Tnode;
     T[]->ch[]=T[]->ch[]=T[];

     FOR(i,,n)
         T[i]=build(T[i-],,n,dfs_list[i]);
     FOR(i,,n) {
         int k=read();
         if(k>R[i]-L[i]+) puts("-1");
         else printf("%d\n",query(T[L[i]-],T[R[i]],,n,k));
     }
     return ;
 }