[K大子串 后缀自动机 模板题] BZOJ 3998 [TJOI2015]弦论

题目大意：分别求 重复子串算一次 重复子串算多次 的 K小子串

若重复子串算一次 那么每个状态计数1次
若重复子串算多次 那么每个状态计数|right(x)|次
然后跑就好了

#include<cstdio>#include<cstdlib>#include<algorithm>using namespace std;inline char nc(){  static char buf[100000],*p1=buf,*p2=buf;  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }  return *p1++;}inline void read(int &x){  char c=nc(),b=1;  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;}inline int read(char *s){  char c=nc(); int len=0;  for (;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;  for (;c>='a' && c<='z';s[++len]=c,c=nc()); s[++len]=0; return len-1;}const int N=500005;struct state{  int len,link,next[26];}st[N<<1];int ncnt,last;int cnt[2][N<<1],size[2][N<<1];inline void Extend(char c){  int cur=++ncnt,p; c-='a';  st[cur].len=st[last].len+1; cnt[1][cur]=1; size[1][cur]=1;  for (p=last;p!=-1 && !st[p].next[c];p=st[p].link)    st[p].next[c]=cur;  if (p==-1)    st[cur].link=0;  else{    int q=st[p].next[c];    if (st[q].len==st[p].len+1)      st[cur].link=q;    else{      int nq=++ncnt;      st[nq].len=st[p].len+1;      st[nq].link=st[q].link;      for (int i=0;i<26;i++) st[nq].next[i]=st[q].next[i];      for (;p!=-1 && st[p].next[c]==q;p=st[p].link)    st[p].next[c]=nq;      st[q].link=st[cur].link=nq;    }  }  last=cur;}int tmp[N<<1],sum[N<<1];inline void Sort(){  for (int i=0;i<=ncnt;i++) sum[st[i].len]++;  for (int i=1;i<=st[last].len;i++) sum[i]+=sum[i-1];  for (int i=0;i<=ncnt;i++) tmp[sum[st[i].len]--]=i;}char A[N];int na;char ans[N]; int len=0;inline void Solve(int *cnt,int *size,int K){  if (K>size[0]){    printf("-1\n"); return;  }  int p=0; len=0;  while (1){    if (K<=cnt[p]) break;    K-=cnt[p];    for (int i=0;i<26;i++)      if (st[p].next[i]){    if (K<=size[st[p].next[i]]){      p=st[p].next[i]; ans[++len]='a'+i;      break;    }else      K-=size[st[p].next[i]];      }  }  ans[++len]=0;  printf("%s\n",ans+1);}int main(){  int T,K;  freopen("t.in","r",stdin);  freopen("t.out","w",stdout);  na=read(A);  st[0].link=-1;  for (int i=1;i<=na;i++) Extend(A[i]);  Sort();  for (int i=ncnt+1;i;i--) if (tmp[i]) cnt[1][st[tmp[i]].link]+=cnt[1][tmp[i]]; cnt[1][0]=0;  for (int i=1;i<=ncnt;i++) cnt[0][i]=1;  for (int i=0;i<=ncnt;i++) size[0][i]=cnt[0][i];  for (int i=ncnt+1;i;i--) for (int j=0;j<26;j++) if (st[tmp[i]].next[j]) size[0][tmp[i]]+=size[0][st[tmp[i]].next[j]];  for (int i=0;i<=ncnt;i++) size[1][i]=cnt[1][i];  for (int i=ncnt+1;i;i--) for (int j=0;j<26;j++) if (st[tmp[i]].next[j]) size[1][tmp[i]]+=size[1][st[tmp[i]].next[j]];  read(T); read(K);  Solve(cnt[T],size[T],K);  return 0;}