RE了几十发,实在没办法了…只好向管理员要数据,然后发现数据规模与题目描述不符…
建立Trie并求出DFS序,同时根据DFS序确定字典序
然后每次询问相当于询问子树第k小,用主席树维护,注意压缩内存
时间复杂度$O(L+n\log w)$,L为所有串长度之和
#include<cstdio>
#include<cstring>
const int N=100010,T=2100000,M=2000010;
int n,q,i,k,v[T],st[T],en[T],tot,dfn,now,root[T],l[M],r[M],cnt,val[M],a[N],b[N],seq[T];
int g[T],nxt[T],to[T],ed;
char s[T],ch[T],w[T];
inline void addedge(int x,int y,int z){to[++ed]=y,w[ed]=z;nxt[ed]=g[x];g[x]=ed;}
inline int son(int x,int y){
for(int i=g[x];i;i=nxt[i])if(w[i]==y)return to[i];
return 0;
}
inline void ins(int p){
for(int x=0,i=0,j=std::strlen(s),w;i<j;i++){
if(!son(x,w=s[i]-'a'))addedge(x,++tot,w);
x=son(x,w);
if(i==j-1)v[x]=p;
}
}
void dfs(int x){
if(v[x])a[v[x]]=++now;
seq[st[x]=++dfn]=a[v[x]];
for(int i=0;i<26;i++)if(son(x,i))dfs(son(x,i));
en[x]=dfn;
}
int add(int x,int a,int b,int c){
int y=++cnt;
val[y]=val[x]+1;
if(a==b)return y;
int mid=(a+b)>>1;
if(c<=mid)l[y]=add(l[x],a,mid,c),r[y]=r[x];else l[y]=l[x],r[y]=add(r[x],mid+1,b,c);
return y;
}
inline int ask(int k){
int x=0,i=0,j=std::strlen(ch),w,y,c=1,d=n,mid;
for(;i<j;x=son(x,w),i++)if(!son(x,w=ch[i]-'a'))return -1;
y=root[st[x]-1],x=root[en[x]];
if(val[x]-val[y]<k)return -1;
while(c<d){
mid=(c+d)>>1,j=val[l[x]]-val[l[y]];
if(k<=j)d=mid,x=l[x],y=l[y];else k-=j,c=mid+1,x=r[x],y=r[y];
}
return b[c];
}
int main(){
scanf("%d%d",&n,&q);
for(i=1;i<=n;i++)scanf("%s",s),ins(i);
for(dfs(0),i=1;i<=n;i++)b[a[i]]=i;
for(i=1;i<=dfn;i++)root[i]=seq[i]?add(root[i-1],1,n,seq[i]):root[i-1];
while(q--)scanf("%d%s",&k,ch),printf("%d\n",ask(k));
return 0;
}