Kattis - repeatedsubstrings(后缀数组)

要求字符串中找出两个最大相同的公共子串
思路：
利用height数组，height数组最大的位置即为结果，另可通过sa[i]求取最大的位置。

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
using namespace std;
/*
	后缀数组模板，下标从0开始
	s[]原字符串 
	后缀sa[i]和sa[j] LCP(i, j) = min{height[k] | min{i, j} < k <= max{i, j}}最长公共前缀 
	Heigth[i] : 表示Suffix[SA[i]]和Suffix[SA[i - 1]]的最长公共前缀，
		也就是排名相邻的两个后缀的最长公共前缀	
	rank[i] 第i个后缀的排名; SA[i] 排名为i的后缀位置
*/ 

const int N=1000010;
char s[N];
int c[N],x[N],y[N],n,m=256,height[N],rk[N],sa[N];
void Suffix()
{
     for(int i=0;i<m;i++) c[i]=0;
     for(int i=0;i<n;i++) c[x[i]=s[i]]++;
     for(int i=1;i<m;i++) c[i]+=c[i-1];
     for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
     for(int k=1;k<=n;k<<=1)
     {
         int p=0;
         for(int i=n-k;i<n;i++) y[p++]=i;
         for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
 
         for(int i=0;i<m;i++) c[i]=0;
         for(int i=0;i<n;i++) c[x[y[i]]]++;
         for(int i=1;i<m;i++) c[i]+=c[i-1];
         for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
         swap(x,y);
         p=1;x[sa[0]]=0;
         for(int i=1;i<n;i++)
            x[sa[i]]= y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
         if(p>=n) break;
         m=p;
     }
}
void getheight()
{
    for(int i=0;i<n;i++) rk[sa[i]]=i;
    for(int i=0,k=0;i<n;i++)
    {
        if(rk[i])
        {
            if(k) --k;
            //else k=0;
            int j=sa[rk[i]-1];
            while(s[i+k]==s[k+j]) k++;
            height[rk[i]]=k;
        }
    }
}

int main()
{

	scanf("%s",s);
	
	n = strlen(s);
	s[n++] = '0';
    Suffix();
    getheight();
    int res = sa[0],len=1;
	for(int i=1;i<n;i++)
	{
		if(height[i]>len){
			len = height[i];
			res = sa[i];
		}
	}    
	for(int i=res;i<=res+len-1;i++)
		printf("%c",s[i]);
}