POJ 2774 Long Long Message & SP1811 LCS - Longest Common Substring 题解

POJ：题目传送门
洛谷：题目传送门

题目大意： 求两个字符串的最长公共子串长度。

题解

后缀数组入门题，将两个字符串接在一起，中间用一个字符集以外的字符隔开，然后求出 $height$ 数组，找到一个最大的 $height[i]$，满足 排名为 $i$ 和 $i-1$ 的两个后缀的起点分别在两个字符串中即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 500010

int n,m,len1,ans;
int fir[maxn],sec[maxn],sa[maxn],c[maxn<<1],rk[maxn],height[maxn];
char s[maxn<<1],ss[maxn];
void get_sa()
{
	m=200;
	for(int i=1;i<=n;i++)c[i]=0;
	for(int i=1;i<=n;i++)c[fir[i]=s[i]]++;
	for(int i=2;i<=m;i++)c[i]+=c[i-1];
	for(int i=n;i>=1;i--)sa[c[fir[i]]--]=i;
	for(int k=1;k<n;k<<=1)
	{
		int tot=0;
		for(int i=n-k+1;i<=n;i++)sec[++tot]=i;
		for(int i=1;i<=n;i++)if(sa[i]>k)sec[++tot]=sa[i]-k;
		
		for(int i=1;i<=m;i++)c[i]=0;
		for(int i=1;i<=n;i++)c[fir[i]]++;
		for(int i=2;i<=m;i++)c[i]+=c[i-1];
		for(int i=n;i>=1;i--)sa[c[fir[sec[i]]]--]=sec[i],sec[i]=0;
		
		for(int i=1;i<=n;i++)swap(fir[i],sec[i]);
		fir[sa[1]]=tot=1;
		for(int i=2;i<=n;i++)
		if(sec[sa[i]]==sec[sa[i-1]]&&sec[sa[i]+k]==sec[sa[i-1]+k])fir[sa[i]]=tot;
		else fir[sa[i]]=++tot;
		
		if(tot==n)break;m=tot;
	}
}
void get_height()
{
	int k=0;
	for(int i=1;i<=n;i++)rk[sa[i]]=i;
	for(int i=1;i<=n;i++)
	{
		if(rk[i]==1){height[rk[i]]=0;continue;}
		if(k)k--; int j=sa[rk[i]-1];
		while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])k++;
		height[rk[i]]=k;
	}
}
bool belong(int x){return x<=len1;}

int main()
{
	scanf("%s",ss+1);//POJ那题的话改改输入就好，那题是多组输入
	len1=strlen(ss+1);n=ans=0;
	for(int i=1;i<=len1;i++)s[++n]=ss[i];
	s[++n]='#';
	scanf("%s",ss+1);m=strlen(ss+1);
	for(int i=1;i<=m;i++)s[++n]=ss[i];
	get_sa();get_height();
	for(int i=2;i<=n;i++)
	if(height[i]>ans&&belong(sa[i])!=belong(sa[i-1]))ans=height[i];
	printf("%d\n",ans);
}