HDU 1403 Longest Common Substring(后缀数组入门)

传送门:http://acm.hdu.edu.cn/showproblem.php?pid=1403

拖了这么久的后缀数组，终于准备学一学了。
做的第一题（套的第一个板子），虽然还是没看不懂是怎么实现的，但是还是可以套一套的。

suffix(i)表示[0,len)的字符串的一个后缀，下标为[i,len)。

后缀数组里，要求的是三个数组，sa[],rank[],height[]。
sa[i]表示的排名为i的后缀的位置。范围[1, len]。
rank[i]表示的是suffix(i)是排名第几。范围[0, len)。
sa[i]和rank[i]是一个相对关系。
height[i]表示suffix(sa[i-1])和suffix(sa[i])的最长公共前缀。范围[2, len]。

这题是求两个字符串的最长公共子串。而每个字符串的子串都是这个字符串的某个后缀的某个前缀。所以我们求出height数组以后，可以知道，公共前缀比较长的后缀的位置，肯定是排名相邻的，那么，我们只要把两个字符串拼接起来，中间拿一个没出现过的字符隔开。然后遍历height数组，如果两个后缀不属于同一个字符串，就更新答案，找出最长的。

后缀数组有两种实现，一个是倍增算法，复杂度O(nlogn)。一个是DC3算法，复杂度O(n)。
我参考了09年国家集训队论文，用的是倍增算法。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <ctime>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define pb push_back
#define mp make_pair
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define calm (l+r)>>1
const int INF = 2139062143;

const int maxn=200010;

struct SA{
    int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
    int rank[maxn],height[maxn];
    int cmp(int *r,int a,int b,int l){
        return r[a]==r[b]&&r[a+l]==r[b+l];
    }
    void da(int *r,int *sa,int n,int m){
         int i,j,p,*x=wa,*y=wb,*t;
         for(i=0;i<m;i++) ws[i]=0;
         for(i=0;i<n;i++) ws[x[i]=r[i]]++;
         for(i=1;i<m;i++) ws[i]+=ws[i-1];
         for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
         for(j=1,p=1;p<n;j*=2,m=p){
           for(p=0,i=n-j;i<n;i++) y[p++]=i;
           for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
           for(i=0;i<n;i++) wv[i]=x[y[i]];
           for(i=0;i<m;i++) ws[i]=0;
           for(i=0;i<n;i++) ws[wv[i]]++;
           for(i=1;i<m;i++) ws[i]+=ws[i-1];
           for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
           for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
           x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
         }
         return;
    }
    void calheight(int *r,int *sa,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];r[i+k]==r[j+k];k++);
         return;
    }
}rst;
char str[maxn];
int r[maxn],sa[maxn];
int main(){
    //freopen("D://input.txt","r",stdin);
    while(scanf("%s",str)!=EOF){
        int len=strlen(str);
        int mid=len;
        str[len]='#';
        scanf("%s",str+len+1);
        len=strlen(str);
        for(int i=0;i<=len;i++){
            r[i]=str[i];
        }
        rst.da(r,sa,len+1,300);
        rst.calheight(r,sa,len);
        int ans=0;
        for(int i=2;i<=len;i++){
            if(ans<rst.height[i]){
                if((sa[i]>mid&&sa[i-1]<mid)||(sa[i]<mid&&sa[i-1]>mid)){
                    ans=rst.height[i];
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}