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

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

#include<cstdio>#include<cstring>#include<iostream>#include<vector>#define maxn 100021using 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;}