后缀数组应用——公共子串

最长公共子串(poj2774)

给出两个字符串，求出他们的最长公共子串

寻常的做法是 O( n )，或者 O( nlogn )。用后缀数组我们可以做到 O( n )，非常高效。首先我们知道任意一个子串都是原串的某个后缀的一个前缀，那么我们把两个串接在一起，中间用一个没有出现过的字符做分隔符。问题就转变成了这个新字符串的后缀的最长前缀。那么是不是只要找 Height 数组中最大的就可以了？？不一定！！因为两个后缀可能在同一个原串中，特别注意一下就好了。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAX_N = 200005;

int a[MAX_N], sa[MAX_N], n, la, lb, r[MAX_N], h[MAX_N];
int ws[MAX_N], wv[MAX_N], wa[MAX_N], wb[MAX_N];
char s[MAX_N >> 1];

void da(int *a, int *sa, int n, int m)
{
	int *x = wa, *y = wb;
	for (int i = 0; i < m; i ++) ws[i] = 0;
	for (int i = 0; i < n; i ++) ws[x[i] = a[i]] ++;
	for (int i = 1; i < m; i ++) ws[i] += ws[i - 1];
	for (int i = n - 1; i >= 0; i --) sa[-- ws[x[i]]] = i;
	for (int k = 1; k <= n; k <<= 1){
		int p = 0;
		for (int i = n - k; i < n; i ++) y[p ++] = i;
		for (int i = 0; i < n; i ++) if (sa[i] >= k) y[p ++] = sa[i] - k;
		for (int i = 0; i < n; i ++) wv[i] = x[y[i]];
		for (int i = 0; i < m; i ++) ws[i] = 0;
		for (int i = 0; i < n; i ++) ws[wv[i]] ++;
		for (int i = 1; i < m; i ++) ws[i] += ws[i - 1];
		for (int i = n - 1; i >= 0; i --) sa[-- ws[wv[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 - 1]] == y[sa[i]]) && (y[sa[i - 1] + k] == y[sa[i] + k]) ? p - 1 : p ++;
		if (p >= n) break; m = p;
	} 
}

void calc()
{
	for (int i = 1; i <= n; i ++) r[sa[i]] = i;
	int k = 0, j;
	for (int i = 0; i < n; h[r[i ++]] = k)
		for (k ? k -- : 0, j = sa[r[i] - 1]; a[i + k] == a[j + k]; k ++);
}

void init()
{
	scanf("%s", s); la = strlen(s);
	for (int i = 0; i < la; i ++)
		a[i] = (int)s[i];
	a[la] = 1;
	scanf("%s", s); lb = strlen(s);
	for (int i = la + 1; i < la + 1 + lb; i ++)
		a[i] = (int)s[i - la - 1];
	n = la + lb + 1;
	a[n] = 0;
	da(a, sa, n + 1, 128); 
	calc();
}

void doit()
{
	int ans = -1;
	for (int i = 2; i <= n; i ++){
		if ((sa[i] > la && sa[i - 1] < la) || (sa[i] < la && sa[i - 1] > la))
			if (h[i] > ans) ans = h[i];
	}
	printf("%d\n", ans);
}

int main()
{
	init();
	doit();
	return 0;
}