藍橋杯 2014A組決賽第五題 重複模式 （後綴數組）

作爲 drd 的好朋友，技術男 atm 在 drd 生日時送給他一個超長字符串 S 。atm 要 drd 在其中找出一個最長的字符串 T ，使得 T 在 S 中至少出現了兩次，而他想說的祕密就藏在 T 中。
由於字符串實在是太長了，drd 總是找不到合適的 T 。於是 drd 請你幫他找到這個 T 的長度。
【輸入格式】
一行。一個字符串，即題目中說的S 。
【輸出格式】
一行。一個整數，表示最長的 T 的長度。
【樣例輸入】
ababa
【樣例輸出】
3
「數據範圍」
對於 30% 的數據，S長度 <= 100
對於 60% 的數據，S長度 <= 8000
對於 100% 的數據，S長度 <= 500000
資源約定：
峯值內存消耗 < 256M
CPU消耗 < 1000ms

#include<iostream>
#include<cstring>
#include<algorithm>
#define max 1000000
using namespace std;
char str[max],*p[max];
 int Rank[max], sa[max],wa[max],wb[max],c[max],height[max];
void DA(int n,int m)
{
	int *x = wa, *y = wb;
	int i, j, p;
	for (i = 0;i < n;i++)       c[x[i] = str[i] - 'A'] += 1;
	for (j = 1;j < m;j++)       c[j] += c[j - 1];
	for (i = n - 1;i >= 0;i--)  sa[--c[x[i]]] = i;    //第一次排序
	for (p = 1, j = 1, i = 0;p <n;j = j * 2)
	{
		for (p = 0, i = n - 1;i >=n-j;i--)
	        y[p++] = i;
		for (i = 0;i < n;i++) if (sa[i] >= j) y[p++] = sa[i] - j;
		memset(c, 0, sizeof(int)*m);
		for ( i = 0;i<n;++i) c[x[y[i]]]++;
		for ( i = 1;i<m;++i) c[i] += c[i - 1];
		for ( i = 0;i<n;++i) sa[--c[x[y[i]]]] = y[i];
		swap(x, y);
		x[sa[0]] = 0; p = 1;
		for ( i = 1;i<n;++i)
			x[sa[i]] = (y[sa[i]] == y[sa[i - 1]] && y[sa[i] + j] == y[sa[i - 1] + j]) ? p -1 : p++;
		m = p;
	}
}
void calheight(char *r, int n)
{
	int i, j, k = 0;
	for (i = 1;i <= n;i++) Rank[sa[i]] = i;
	for (i = 0;i < n;height[Rank[i++]] = k)
		for (k?--k:0,j = sa[Rank[i] - 1];i+k<n&&j+k<n&&r[i + k] == r[j + k];k++); //求h[i] = height[rank[i]];
}
	int main()
	{
		freopen("in.txt", "r", stdin);
		int i = 0;
		int ans = 0;
		while (scanf("%c", &str[i++]), str[i - 1] != '\0');
		int z = strlen(str);
		DA(z, 59);
		calheight(str, z);
		for (i = 0;i < z;i++)
		{
			int num = 0;
			ans = ans < height[i] ? height[i]: ans;
			
		}
		cout << z << endl;
		cout << ans;
		return 0;
	}