【bzoj3439】KPM的MC密码

这题乍一看后缀相等很烦的样子……

其实如果把字符串倒过来，那么相等的后缀就可以转化成前缀，前缀相等扔进trie就可以了。

剩下无非是Trie的树链kth，主席树随便维护就好。

注意一个串彻底结束才能打val++。而且因为主席树维护的权值是出现次数，连离散化都不用的。

#include<bits/stdc++.h>
#define N 500010
using namespace std;
int cnt,ch[N][],val[N],in[N],out[N],tpos[N],n;
int sumv[N<<],ls[N<<],rs[N<<];char s[N];
vector<int> q[N];
void ins(int x,char *s){
    int len=strlen(s);int now=,u=;
    for(int i=len-;i>=;i--,now=ch[now][u]){
        u=s[i]-'a';
        if(!ch[now][u])ch[now][u]=++cnt;
    }
    val[now]++;q[now].push_back(x);
}
void dfs(int x){
    for(int i=;i<val[x];i++)in[q[x][i]]=cnt+;
    for(int i=;i<val[x];i++)tpos[++cnt]=q[x][i];
    for(int i=;i<;i++)if(ch[x][i])dfs(ch[x][i]);
    for(int i=;i<val[x];i++)out[q[x][i]]=cnt;
}
void build(int x,int y,int l,int r,int q){
    sumv[x]=sumv[y]+;
    if(l==r)return;int mid=(l+r)>>;
    if(q<=mid){ls[x]=++cnt;rs[x]=rs[y];build(ls[x],ls[y],l,mid,q);}
    else{ls[x]=ls[y];rs[x]=++cnt;build(rs[x],rs[y],mid+,r,q);}
}
int query(int l,int r,int x,int y,int v){
    if(l==r)return l;
    int mid=(l+r)>>;int t=sumv[ls[y]]-sumv[ls[x]];
    if(t>=v)return query(l,mid,ls[x],ls[y],v);
    else return query(mid+,r,rs[x],rs[y],v-t);
}
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
int main(){
    n=read();
    for(int i=;i<=n;i++)scanf("%s",s),ins(i,s);
    cnt=;dfs();cnt=n+;
    for(int i=;i<=n;i++)build(i+,i,,n,tpos[i]);
    for(int i=;i<=n;i++){
        int x=read();
        if(sumv[out[i]+]-sumv[in[i]]<x)puts("-1");
        else printf("%d\n",query(,n,in[i],out[i]+,x));
    }
    return ;
}