POJ2217 Secretary 后缀数组&&高度数组

学后缀数组后的一道裸题。先来讲讲收获，作为字符串初学者，后缀数组也是刚刚在学，所幸的是有一篇好的论文《后缀数组－－处理字符串的有力工具》by 罗穗骞，里面非常详尽地介绍了有关后缀数组的概念，也就是sa[i]和rk[i]表示的是什么。理解了它们互为逆运算后就不难理解sa[rk[i]]=i rk[sa[i]]=i,所以知道其中一个就可以知道另外一个，然后学习了一下后缀数组中的倍增算法构建，虽然效率比不上线性的算法，但是胜在好理解，好写。当然大神的代码我是不怎么懂，我就翻阅了另一本书《挑战程序设计竞赛》里的代码，总之就是两个材料一起看，搞懂了什么是高度数组lcp, 也知道了h[i]>=h[i-1]-1,理解这个不等式我觉得挺关键的，利用这一点就可以写出一个线性的算法求出高度数组。高度数组的一个应用就是求两个串的最长公共子串。对于S1和S2，我们构造一个新串S1#S2,然后建立后缀数组，求出lcp，可以证明的是，最长公共前缀一定是出现在后缀数组中相邻的两个子串里的（否则越离越远，只可能更小）。利用这一点，我们找出满足sa[i]和sa[i+1]分属于S1,S2的串，然后对应的lcp[i]的最大值就是答案。代码完全参(zhao)考(chao)挑战程序设计竞赛,写下来权当学习。。

#pragma warning(disable:4996)
#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
#include<cmath>
#include<cstdio>
#define maxn 500000
using namespace std;

int rk[maxn+50], sa[maxn+50],lcp[maxn+50];
int tmp[maxn + 50];
int k,n;
bool cmp_sa(int i, int j)
{
	if (rk[i] != rk[j]) return rk[i] < rk[j];
	else {
		int ri = i + k <= n ? rk[i + k] : -1;
		int rj = j + k <= n ? rk[j + k] : -1;
		return ri < rj;
	}
}

void construct_sa(char *s, int *sa)
{
	n = strlen(s);
	for (int i = 0; i <= n; i++){
		sa[i] = i;
		rk[i] = i < n ? s[i] : -1;
	}
	for (k = 1; k <= n; k <<= 1)
	{
		sort(sa, sa + n + 1, cmp_sa);
		tmp[sa[0]] = 0;
		for (int i = 1; i <= n; i++)
		{
			tmp[sa[i]] = tmp[sa[i - 1]] + (cmp_sa(sa[i - 1], sa[i]) ? 1 : 0);
		}
		for (int i = 0; i <= n; i++){
			rk[i] = tmp[i];
		}
	}
}

void construct_lcp(char *s, int *sa, int *lcp)
{
	n = strlen(s);
	for (int i = 0; i <= n; i++) rk[sa[i]] = i;
	int h = 0;
	lcp[0] = 0; 
	for (int i = 0; i < n; i++){
		int j = sa[rk[i] - 1];
		for (h ? h-- : 0; j + h < n&&i + h < n&&s[j + h] == s[i + h]; h++);
		lcp[rk[i] - 1] = h;
	}
}

char str1[maxn + 50], str2[maxn + 50];

int main()
{
	int T; cin >> T; getchar();
	while (T--)
	{
		gets(str1);
		gets(str2);
		int len = strlen(str1);
		str1[len] = '#';
		str1[len + 1] = '';
		strcat(str1, str2);
		construct_sa(str1 ,sa);
		construct_lcp(str1,sa,lcp);
		int ans = 0;
		for (int i = 0; i < strlen(str1); i++)
		{
			if ((sa[i] < len) != (sa[i + 1]) < len){
				ans = max(ans, lcp[i]);
			}
		}
		printf("Nejdelsi spolecny retezec ma delku %d.
", ans);
	}
	return 0;
}