whu1564 求字符串的所有循环子串第k大 ：后缀数组模板

原题少了当前的字符，不过并无差别

利用sa数组可以很方便的求出来==

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct SA{
  int sa[2000005],t[2000005],t2[2000005],c[2000005],n,m;
  void build_sa(char *s,int _n){
    n=2*_n+1; m=128;
    int i,*x=t,*y=t2;
    //基数排序
    for (i=0;i<m;i++) c[i]=0;
    for (i=0;i<n;i++) c[x[i]=s[i]]++;
    for (i=1;i<m;i++) c[i]+=c[i-1];
    for (i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for (int k=1;k<=n;k<<=1){
        int p=0;
        //直接利用sa数组排序第二关键字
        for (i=n-k;i<n;i++) y[p++]=i;
        for (i=0;i<n;i++) if (sa[i]>=k) y[p++]=sa[i]-k;
        //基数排序第一关键字
        for (i=0;i<m;i++) c[i]=0;
        for (i=0;i<n;i++) c[x[y[i]]]++;
        for (i=0;i<m;i++) c[i]+=c[i-1];
        for (i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        //根据sa和y数组计算新的x数组
        swap(x,y);
        p=1; x[sa[0]]=0;
        for (i=1;i<n;i++)
  x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        if (p>=n) break;
        m=p;//下次基数排序的最大值
    }
  }
  int solve(int k){
    int i,cnt=0;
    for (i=0;i<n;i++)
    {
      if (sa[i]<=n/2&&sa[i]!=0) cnt++;
      if (cnt==k) return sa[i];
    }
  }
}H;
char s[2000005];
int main()
{
    int i,n,k;
    while (~scanf("%d%d",&n,&k))
    {
      scanf("%s",s);
      for (i=0;i<n;i++) s[i+n]=s[i]=s[i]-'0';
      s[2*n]=10;
      H.build_sa(s,n);
      printf("%d\n",H.solve(k));
    }
    return 0;
}

题目链接：http://acm.whu.edu.cn/land/problem/detail?problem_id=1564