【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;
}