BZOJ 3439 Kpm的MC密码

倒着建trie，然后主席树来求子树第k大。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define maxn 300500
using namespace std;
int n,trie[maxn][],fath[maxn],cnt=,tag[maxn],k,rot,kr,dfn[maxn],mx[maxn],times=,line[maxn];
int tot=,ls[maxn<<],rs[maxn<<],sum[maxn<<];
char s[maxn];
vector <int> v[maxn],root[maxn];
void insert(int father,int &now,int l,int x)
{
    if (!now) {now=++cnt;fath[now]=father;}
    if (!l) {tag[x]=now;v[now].push_back(x);return;}
    insert(now,trie[now][s[l-]-'a'+],l-,x);
}
void build(int &now,int left,int right)
{
    now=++tot;
    if (left==right) return;
    int mid=left+right>>;
    build(ls[now],left,mid);
    build(rs[now],mid+,right);
}
void modify(int last,int &now,int left,int right,int pos)
{
    now=++tot;sum[now]=sum[last]+;
    if (left==right) return;
    int mid=left+right>>;
    ls[now]=ls[last];rs[now]=rs[last];
    if (pos<=mid) modify(ls[last],ls[now],left,mid,pos);
    else modify(rs[last],rs[now],mid+,right,pos);
}
int ask(int last,int now,int left,int right,int x)
{
    if (sum[now]-sum[last]<x) return -;
    if (left==right) return left;
    int mid=left+right>>,lsum=sum[ls[now]]-sum[ls[last]];
    if (x<=lsum) return ask(ls[last],ls[now],left,mid,x);
    else return ask(rs[last],rs[now],mid+,right,x-lsum);
}
void dfs(int now)
{
    dfn[now]=mx[now]=++times;line[times]=now;
    for (int i=;i<=;i++)
    {
        if (trie[now][i])
        {
            dfs(trie[now][i]);
            mx[now]=max(mx[now],mx[trie[now][i]]);
        }
    }
}
int main()
{
    scanf("%d",&n);
    for (int i=;i<=n;i++)
    {
        scanf("%s",s);
        insert(,rot,strlen(s),i);
    }
    dfs(rot);
    build(kr,,n);root[].push_back();
    for (int i=;i<=times;i++)
    {
        int now=line[i];
        if (!v[now].size()) root[i].push_back(root[i-][root[i-].size()-]);
        else
        {
            root[i].push_back(tot+);modify(root[i-][root[i-].size()-],kr,,n,v[now][]);
            for (int j=;j<v[now].size();j++)
            {
                root[i].push_back(tot+);
                modify(root[i][j-],kr,,n,v[now][j]);
            }
        }
    }
    for (int i=;i<=n;i++)
    {
        scanf("%d",&k);
        int ret=tag[i];
        printf("%d\n",ask(root[dfn[ret]-][root[dfn[ret]-].size()-],root[mx[ret]][root[mx[ret]].size()-],,n,k));
    }
    return ;
}