【BZOJ】Kpm的MC密码-trie树+主席树+dfs序

传送门：BZOJ-Kpm的MC密码

---

题意

    现在定义这么一个概念，如果字符串s是字符串c的一个后缀，那么我们称c是s的一个kpm串。
    系统将随机生成n个由a…z组成的字符串，由1…n编号（s1,s2…,sn），然后将它们按序告诉你，接下来会给你n个数字，分别为k1…kn，对于每一个ki，要求你求出列出的n个字符串中所有是si的kpm串的字符串的编号中第ki小的数，如果不存在第ki小的数，则用-1代替。（比如说给出的字符串是cd,abcd,bcd,此时k1=2，那么”cd”的kpm串有”cd”,”abcd”,”bcd”,编号分别为1,2,3其中第2小的编号就是2）（PS：如果你能在相当快的时间里回答完所有n个ki的查询，那么你就可以成功帮kpm进入MC啦~~）

---

输入

第一行一个整数 n 表示字符串的数目
接下来第二行到n+1行总共n行，每行包括一个字符串，第i+1行的字符串表示编号为i的字符串
接下来包括n行，每行包括一个整数ki，意义如上题所示

---

输出

包括n行，第i行包括一个整数，表示所有是si的kpm串的字符串的编号中第ki小的数

---

数据范围

对于100%的数据，1<=n<=100000

---

题解

就是把每个字符串倒着加入trie树，先dfs搜一遍，再以点建立主席树，查询就好了。

---

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector> 
using namespace std;
const int N=1e5+10;
int n,trie[N*3][26],root=0,cnt;
int data[N],in[N],out[N]; 
vector<int>od[N*3];//order 
char c[N*3];
//数据范围很迷 

struct C_t{
    int sz,ch[2];
}a[N*20];

inline void insert(int k,char c[])
{
    int len=strlen(c)-1;int u=root;
    for(int i=len;i>=0;i--){
        int now=c[i]-'a';
        if(!trie[u][now]){
            trie[u][now]=++cnt;
        }
        u=trie[u][now];
    }
    od[u].push_back(k);
}

inline void dfs(int k)
{
    int qw=od[k].size();
    if(qw){
       for(int i=0;i<qw;i++){
         in[od[k][i]]=cnt+1;
       }    
       for(int i=0;i<qw;i++){
         data[++cnt]=od[k][i];
       }
    }
    for(int i=0;i<26;i++){
        if(trie[k][i]){dfs(trie[k][i]);}
    }
    for(int i=0;i<qw;i++){
        out[od[k][i]]=cnt;
    }
}

inline void build(int now,int l,int r,int rt,int k)
{
    a[rt].sz=a[now].sz+1;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(k<=mid){
        a[rt].ch[1]=a[now].ch[1];
        a[rt].ch[0]=++cnt;
        build(a[now].ch[0],l,mid,a[rt].ch[0],k);
    }else{
        a[rt].ch[0]=a[now].ch[0];
        a[rt].ch[1]=++cnt;
        build(a[now].ch[1],mid+1,r,a[rt].ch[1],k);
    }
}

inline int get(int l,int r,int L,int R,int k)
{
    if(l==r) return l;
    int mid=(l+r)>>1,sum=a[a[R].ch[0]].sz-a[a[L].ch[0]].sz;
    if(k<=sum){
        return get(l,mid,a[L].ch[0],a[R].ch[0],k);
    }else{
        return get(mid+1,r,a[L].ch[1],a[R].ch[1],k-sum);
    }
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
       scanf("%s",c);
       insert(i,c);
    } 
    cnt=0;dfs(root);
    cnt=n+1;
    for(int i=1;i<=n;i++){
       build(i,1,n,i+1,data[i]);//最好+1处理，有0不好办
    }
    for(int k,i=1;i<=n;i++){
        scanf("%d",&k);
        if(a[out[i]+1].sz-a[in[i]].sz<k){printf("-1\n");}
        else{
            printf("%d\n",get(1,n,in[i],out[i]+1,k));
        }
    }
    return 0;
}