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;
}