【BZOJ 3439】Kpm的MC密码 主席树+trie树

所有的串都倒着插然后后缀变前缀，然后用主席树维护dfs序查询子树第k大就好了，唯一就是注意一下肯呢过有重复的串

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#define maxn 100021
using namespace std;
vector<int>g[maxn*20];
int ch[maxn*20][26],in[maxn],out[maxn],tot,cnt,cur[maxn];
int rt[maxn],sum[maxn*20],ls[maxn*20],rs[maxn*20],sz,n;
char s[maxn];
void insert(char* s,int id){
int len=strlen(s),rt=0;
for(int c,i=len-1;i>=0;i--){
c=s[i]-'a';
if(!ch[rt][c])ch[rt][c]=++tot;
rt=ch[rt][c];
}g[rt].push_back(id);
}
void dfs(int u){
if(g[u].size()){
int x=cnt+1;
for(int i=0;i<g[u].size();i++){
in[g[u][i]]=x;
cur[++cnt]=g[u][i];
}
}

for(int i=0;i<26;i++)if(ch[u][i])dfs(ch[u][i]);

if(g[u].size()){
for(int i=0;i<g[u].size();i++)out[g[u][i]]=cnt;
}
}
void build(int x,int& y,int l,int r,int id){
y=++sz;ls[y]=ls[x],rs[y]=rs[x],sum[y]=sum[x]+1;
if(l==r)return;
int mid=l+r>>1;
if(id>mid)build(rs[x],rs[y],mid+1,r,id);
else build(ls[x],ls[y],l,mid,id);
}
int query(int x,int y,int l,int r,int k){
if(sum[y]-sum[x]<k)return -1;
if(l==r)return l;
int mid=l+r>>1;
int s=sum[ls[y]]-sum[ls[x]];
if(s>=k)return query(ls[x],ls[y],l,mid,k);
else return query(rs[x],rs[y],mid+1,r,k-s);
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",s);insert(s,i);
}
dfs(0);
for(int i=1;i<=cnt;i++)build(rt[i-1],rt[i],1,n,cur[i]);
int k;
for(int i=1;i<=n;i++){
scanf("%d",&k);
printf("%d\n",query(rt[in[i]-1],rt[out[i]],1,n,k));
}return 0;
}