[BZOJ3439]Kpm的MC密码

题目大意：
　　给定$n(nleq10^5)$个字符串$s_{1sim n}$，$n$次询问，每次询问所有满足$s_i$是$t$的后缀的$t$中，编号第$k_i$小的$t$的编号。

思路：
　　倒着建字典树，根据DFS序建立主席树，维护每个区间内编号出现次数。

#include<list>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=100001,logN=18,S=26;
int n;
class FotileTree {
    private:
        struct Node {
            int val,left,right,vis;
        };
        Node node[N*logN];
        int sz,new_node(const int &p,const int &id) {
            node[++sz]=node[p];
            node[sz].vis=id;
            return sz;
        }
    public:
        int root[N];
        void modify(int &p,const int &b,const int &e,const int &x,const int &y,const int &id) {
            if(node[p].vis!=id) p=new_node(p,id);
            node[p].val+=y;
            if(b==e) return;
            const int mid=(b+e)>>1;
            if(x<=mid) modify(node[p].left,b,mid,x,y,id);
            if(x>mid) modify(node[p].right,mid+1,e,x,y,id);
        }
        int query(const int &p,const int &q,const int &b,const int &e,const int &k) {
            if(node[q].val-node[p].val<k) return -1;
            if(b==e) return b;
            const int mid=(b+e)>>1;
            if(node[node[q].left].val-node[node[p].left].val>=k) return query(node[p].left,node[q].left,b,mid,k);
            return query(node[p].right,node[q].right,mid+1,e,k-(node[node[q].left].val-node[node[p].left].val));
        }
};
FotileTree t;
class Trie {
    private:
        int pos[N],ch[N][S],par[N],dfn[N],size[N],cnt;
        std::list<int> val[N];
        int sz,new_node(const int &p) {
            par[++sz]=p;
            return sz;
        }
        int idx(const char &c) const {
            return c-'a';
        }
        void push_up(const int &p) {
            size[p]=1;
            for(register int i=0;i<S;i++) size[p]+=size[ch[p][i]];
        }
    public:
        void insert(const char s[],const int &id) {
            int u=0;
            for(register int i=0;s[i];i++) {
                u=ch[u][idx(s[i])]=ch[u][idx(s[i])]?:new_node(u);
            }
            val[pos[id]=u].push_back(id);
            for(;u;u=par[u]) push_up(u);
        }
        void dfs(const int &x) {
            dfn[x]=cnt++;
            if(dfn[x]) t.root[dfn[x]]=t.root[dfn[x]-1];
            for(register std::list<int>::iterator i=val[x].begin();i!=val[x].end();i++) {
                t.modify(t.root[dfn[x]],1,n,*i,1,dfn[x]);
            }
            for(int i=0;i<S;i++) {
                if(ch[x][i]) dfs(ch[x][i]);
            }
        }
        int query(const int &id,const int &k) {
            return t.query(t.root[dfn[pos[id]]-1],t.root[dfn[pos[id]]+size[pos[id]]-1],1,n,k);
        }
};
Trie trie;
char s[N];
int main() {
    n=getint();
    for(register int i=1;i<=n;i++) {
        scanf("%s",s);
        std::reverse(&s[0],&s[strlen(s)]);
        trie.insert(s,i);
    }
    trie.dfs(0);
    for(register int i=1;i<=n;i++) {
        printf("%d
",trie.query(i,getint()));
    }
    return 0;
}