BZOJ3439: Kpm的MC密码

一开始想后缀数组+主席树…仔细想想…既然完全是后缀，那么直接将所有串reverse，塞到trie即可，现在问题就是查询子树里关键点第k大，在trie树上dfs，按照dfs序建一颗主席树即可。
友情提示有重复串！！！！
（当模板题水一发就好啦…）

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ls(tr) tree[tr].l
#define rs(tr) tree[tr].r
//by:MirrorGray
using namespace std;
const int N=211111;
char s[N];
int n,cnt,tot=-1,P,sum;
int head[N],ver[N],nxt[N];
int a[N],bi[N],en[N],root[N];
struct Trie{int nxt[26];}tr[N];
struct Segment_tree{int l,r,sum;}tree[N*21];

void add(int x,int y){
    nxt[++tot]=head[x];
    head[x]=tot;
    ver[tot]=y;
}

void insert(int x){
    scanf("%s",s+1);int n=strlen(s+1),now=0;
    reverse(s+1,s+1+n);
    for(int i=1;s[i];i++){
        int &t=tr[now].nxt[s[i]-'a'];
        if(!t)t=++cnt;now=t;
    }
    a[x]=now;add(now,x);
}

void insert(int x,int &y,int al,int ar,int wei,int d){
    tree[y=++cnt]=tree[x];
    if(al==ar){
        tree[y].sum+=d;
        return ;
    }
    int mid=(al+ar)>>1;
    if(wei>mid)insert(rs(x),rs(y),mid+1,ar,wei,d);
    else insert(ls(x),ls(y),al,mid,wei,d);
    tree[y].sum=tree[ls(y)].sum+tree[rs(y)].sum;
}

void dfs(int x){
    bi[x]=tot+1;
    for(int i=head[x];~i;i=nxt[i])tot++,insert(root[tot-1],root[tot],1,n,ver[i],1);
    for(int i=0;i<26;i++)if(tr[x].nxt[i])dfs(tr[x].nxt[i]);
    en[x]=tot;
}

void query(int x,int y,int al,int ar,int &k){
    if(al==ar){
        P=al;k-=tree[y].sum-tree[x].sum;
        return ;
    }
    int mid=(al+ar)>>1;
    if(tree[ls(y)].sum-tree[ls(x)].sum>=k)query(ls(x),ls(y),al,mid,k);
    else query(rs(x),rs(y),mid+1,ar,k-=(tree[ls(y)].sum-tree[ls(x)].sum));
}

int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)insert(i);
    tot=cnt=0;dfs(0);
    for(int i=1;i<=n;i++){
        int k;scanf("%d",&k);
        query(root[ bi[a[i]]-1 ],root[ en[a[i]] ],1,n,k);
        if(k)puts("-1");else printf("%d\n",P);
    }
    return 0;
}