【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
#include
#include
#include
#include 
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;i1;
       }    
       for(int i=0;ifor(int i=0;i<26;i++){
        if(trie[k][i]){dfs(trie[k][i]);}
    }
    for(int i=0;iinline 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]].szprintf("-1\n");}
        else{
            printf("%d\n",get(1,n,in[i],out[i]+1,k));
        }
    }
    return 0;
}

你可能感兴趣的:(主席树,trie树)