HDU1403（後綴數組模板）

問題鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=1403

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN = 200200;
char str[MAXN];
int t1[MAXN], t2[MAXN], sum[MAXN], Rank[MAXN], sa[MAXN];
int height[MAXN], m = 300;
//str[i]存儲字符串， t1[i]存儲第一關鍵字， t2[i]存儲第二關鍵字， sum[i]基數排序輔助數組, 存儲小於i的元素有多少個。 
//sa[i]排名爲i的後綴的位置（sa[排名] = 下標） ，  Rank[i] 第i個後綴的排名（Rank[下標] = 排名 ） 
//height[i] 排名爲i的後綴（sa[i]）與排名爲(i-1)的後綴（sa[i-1]）的LCP 
//m基數排序的最大值 
void getsa()
{
 int n = strlen(str);
 int p, *x = t1, *y = t2;
 memset(sum, 0, sizeof(sum));//輔助數組初始化 
 for(int i = 0; i < n; i++) sum[x[i] = str[i]]++;
 for(int i = 1; i < m; i++) sum[i] += sum[i-1];
 for(int i = n-1; i >= 0; i--) sa[--sum[x[i]]] = i; //以上三行爲第一次排序 
 for(int j = 1; j <= n; j<<=1)
 {
  p = 0;
  // 依靠sa對第二關鍵字排序 
  for(int i = n-j; i < n; i++) y[p++] = i;
  for(int i = 0; i < n; i++) if(sa[i] >= j) y[p++] = sa[i] - j; //以上兩行爲第二關鍵字的排序過程 
  memset(sum, 0, sizeof(sum));
  for(int i = 0; i < n; i++) sum[x[y[i]]]++; //根據第二關鍵字的結果對第一關鍵字排序（firsecsort[i] = x[y[i]];) sum[x[y[i]]] = sum[firsecsort[i]];
  for(int i = 1; i < m; i++) sum[i] += sum[i-1];
  for(int i = n-1; i >= 0; i--) sa[--sum[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]+j] == y[sa[i-1]+j])?p-1:p++;//判斷Rank[i]與Rank[i-1]是否相同 
  if(p>=n) break;                                                                                
  m = p; //下次基數排序的最大值 
 }
}
void getheight()
{
 int j, k = 0;
 int n = strlen(str);
 for(int i= 0; i <= n; i++) Rank[sa[i]] = i;  //求Rank 
 for(int i = 0; i < n; i++)                           
 {
  if(k) k--;
  j = sa[Rank[i]-1];
  while(str[i+k] == str[j+k]) k++;
  height[Rank[i]] = k;
 }
}
int main()
{
 while(~scanf("%s", str))
 {
  int len = strlen(str);
  str[len] = '#';
  scanf("%s", str+len+1);
  int n = strlen(str);
  getsa();
  getheight();
  int ans = 0;
  for(int i = 1; i < n; i++)
  {
   if(height[i] > ans&&((sa[i] < len&&sa[i-1] > len)||(sa[i]>len&&sa[i-1]<len)))
    ans = height[i];
  }
  printf("%d\n", ans);
 }
 return 0;
}