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+100;

ll read() {
    char c=getchar();
    ll f=1,x=0;
    while(!isdigit(c)) {
        if(c=='-') f=-1; c=getchar();
    }
    while(isdigit(c))
        x=x*10+c-'0',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][26],sz;
    vector<int> num[N];
    Trie() {
        sz=0;
        memset(ch,0,sizeof(ch));
    }
    void insert(char *s,int val) {
        int u=0;
        for(int i=0;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+1;
        FOR(i,0,(int)num[u].size()-1) {
            dfs_list[++dfsc]=num[u][i];
            L[num[u][i]]=tmp;
        }
        FOR(c,0,25) if(ch[u][c])
            dfs(ch[u][c]);
        FOR(i,0,(int)num[u].size()-1)
            R[num[u][i]]=dfsc;
    }
} trie;

struct Tnode {
    Tnode *ch[2];
    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[1<<15];
        mempool=G+(1<<15);
        memset(G,0,sizeof(Tnode)*(1<<15));
    }
    G->ch[0]=l,G->ch[1]=r,G->num=x;
    return G++;
}

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

int main()
{
    n=read();
    FOR(i,1,n) {
        scanf("%s",s);
        int n=strlen(s);
        for(int j=0;j*2<n;j++)
            swap(s[j],s[n-1-j]);
        trie.insert(s,i);
    }
    trie.dfs(0);
    
    T[0]=new (0x0,0x0,0) Tnode;
    T[0]->ch[0]=T[0]->ch[1]=T[0];
    
    FOR(i,1,n)
        T[i]=build(T[i-1],1,n,dfs_list[i]);
    FOR(i,1,n) {
        int k=read();
        if(k>R[i]-L[i]+1) puts("-1");
        else printf("%d
",query(T[L[i]-1],T[R[i]],1,n,k));
    }
    return 0;
}