【BZOJ3439】Kpm的MC密码，trie树+dfs序+主席树

Time:2016.05.07
Author:xiaoyimi
转载注明出处谢谢

传送门
思路：
1.have a trie，还得是倒着建的，记录每个结尾节点的id（可能会有重复，所以开一个vector记录一下）
2.对trie树进行dfs序，记录结尾节点的子树区间
3.建立主席树（由于权值就是id标号，所以不用离散化了）
4.1-n依次查询即可
注意：
dfs序在遇到结尾节点时才++记录序号的变量cnt，而且相同串的dfs序号是一样的（就是说一个结尾节点可能包含好多个不同编号的串），左区间取最前的一个（比如dfs到x节点，编号为2，x节点存着两个串，那么它们的左区间都为2，而序号要加两次）
代码：

#include<bits/stdc++.h>
using namespace std;
#define M 100004
#define ls(x) a[x].ch[0]
#define rs(x) a[x].ch[1]
int n,m,cnt,root=M*3-10;
int trie[M*3][26],num[M*3],Ls[M],Rs[M],data[M];
vector <int> q[M*3];
char s[M*3];
struct Chairman_tree
{
    int siz,ch[2];
}a[M*18];
int in()
{
    int t=0;char ch=getchar();
    while (!isdigit(ch)) ch=getchar();
    while (isdigit(ch)) t=(t<<3)+(t<<1)+ch-48,ch=getchar();
    return t;
}
void build(int now,int L,int R,int rt,int val)
{
    a[rt].siz=a[now].siz+1;
    if (L==R) return;
    int mid=(L+R)>>1;
    if (mid>=val)
        rs(rt)=rs(now),
        ls(rt)=++cnt,
        build(ls(now),L,mid,ls(rt),val);
    else
        ls(rt)=ls(now),
        rs(rt)=++cnt,
        build(rs(now),mid+1,R,rs(rt),val);
}
int get(int begin,int end,int L,int R,int k)
{
    if (begin==end) return end;
    int mid=(begin+end)>>1,t=a[ls(R)].siz-a[ls(L)].siz;
    if (k<=t)
        return get(begin,mid,ls(L),ls(R),k);
    else
        return get(mid+1,end,rs(L),rs(R),k-t);
}
void insert(int id,char s[])
{
    int len=strlen(s),now=root;
    for (int i=len-1;i>=0;i--)
    {
        if (!trie[now][s[i]-'a']) trie[now][s[i]-'a']=++cnt;
        now=trie[now][s[i]-'a'];
    }
    num[now]++;
    q[now].push_back(id);
}
void dfs(int x)
{
    if (num[x])
    {       
        for (int i=0;i<num[x];i++)
            Ls[q[x][i]]=cnt+1;
        for (int i=0;i<num[x];i++)
            data[++cnt]=q[x][i];
    }
    for (int i=0;i<26;i++)
        if (trie[x][i]) dfs(trie[x][i]);
    for (int i=0;i<num[x];i++)
        Rs[q[x][i]]=cnt;
}
main()
{
    n=in();
    for (int i=1;i<=n;i++)
        scanf("%s",s),
        insert(i,s);
    cnt=0;
    dfs(root);
    cnt=n+1;
    int k;
    for (int i=1;i<=n;i++) build(i,1,n,i+1,data[i]);
    for (int i=1;i<=n;i++)
    {
        k=in();
        if (a[Rs[i]+1].siz-a[Ls[i]].siz<k) printf("-1\n");
        else printf("%d\n",get(1,n,Ls[i],Rs[i]+1,k));
    }
}