BZOJ3448 : [Usaco2014 Feb]Auto-complete

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;

}