刘汝佳蓝书后缀数组模板解释及补全

      相信很多初学后缀数组的ACMer在学习蓝书中的后缀数组部分遇到了一些障碍，可能像我一样看明白了P219 ——220的讲解和算法，百度了基数排序的方法，然后被卡在P221的代码上了，本文目的即分享我对这段代码的理解。

      首先明确其中每个变量的含义，n为字符串s的长度，m为字符串中可能包含的最大字符值+1（故一开始应置为'z' + 1），sa数组存的是P220四幅图中每一幅结束时最下面一行的最终排名对应情况（！！“排名对应情况”看不懂下面还有解释），y数组存的是后三幅图中二元组的第二元的排名对应情况（第一幅没有第二元），x[i]表示第i种字符/二元组对应的编号，c[i]表示第0种到第i种字符/二元组的总数量。

      先解释清楚排名对应情况的含义，以P220第二幅图为例，最终排名结果为1 2 4 1 1 1 2 3 ，那么他的排名对应情况为0 3 4 5 1 6 7 2，这是怎么对应的呢？就是先把排名第1的位置下标放在最前面，也就是0 3 4 5，然后再把排第2的下标放前面，就是1 6 ，然后把排第三......

     下面只以k第一次循环时的情况说明整段代码。

     33 到37行用x数组统计各类字符的种类，用c数组统计每种字符出现的次数，然后用sa数组把他们依次装入，装的时候就是基数排序从木桶里将每一个数拿出来然后放到相应的位置的过程，装完以后sa的内容就是前文所说的第一幅图中最后一行的排名对应情况，此时0 —— 7 七个下标正好在sa中各出现一次。

     接下来进入了k的循环，k的含义是什么呢，以第二幅图为例，第二行每个二元组头上都有条竖线和一条斜线，斜线所指的位置比竖线指的大k，对，k就是这个意思，可以代入第三四幅图中，也就是k循环的二三次，来看看是不是这样。

    下面要排序二元组的第二关键字，然后把排名装入y数组，我们看第二幅图的中间一行，二元组的第二关键字从左至右依次为 1 2 1 1 1 1 2 0，所以排名对应情况为7 0 2 3 4 5 1 6，那么怎么实现这个排序呢？代码39——41行，为的是把0对应的位置先装进y数组里，接下来42——45，是把剩下的放进y数组。42——45是怎么做到把剩下的放进去的？看P220的第二幅图，第一行的数字就是第一幅图的最后一行数字，而第一幅图最后一行数字的排名对应情况已经放到sa数组里，sa[i] - k表示的是一种对应关系：第二幅图的第一行里第i个数字，对应到第二行的第i - k个数字的第二元，只要明白了这种对应关系，代码为什么这么写，琢磨一下就可以明白了。

    第46到第52行是排序第一关键字，这里面比较难理解的是51——52行，首先要倒着循环是因为i大的y[i]是第二元比较大的，所以第一元相同时要先放第二元比较大的，因此要倒着循环。那么第y[i]个数应该放哪呢？从第二幅图中看到，第二行中第y[i]个二元组对应的第一元是第一行中的第y[i]个数，第一行中第y[i]个数的排名就是当前的sa[- -c[x[y[i]]]]（为什么？x[y[i]]表示y[i]的种类，c[x[y[i]]]表示放到第x[y[i]]种已经放了多少个，不断- -使得大小相同的x[y[i]]排出先后，所以第y[i]个二元组放在sa里是sa[--c[x[y[i] ] ] ]）。

   54——58行中，p代表的是不同的二元组的数量，如果p >= n，则build_sa结束，现在sa中已经排完序，所以相等的二元组在sa里只可能相邻，56行为的就是判断有多少相邻且不等的二元组，这样来统计这样排序之后不同的二元组的数量p，59行如果算法未结束将二元组的种类m更新为p。

至此，build_sa部分的代码解释完了，其他部分的代码相对来说要更好理解，先给出补全版的全部代码。

#pragma warning(disable:4786)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<string>
#include<sstream>
#define LL long long
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define mem(a,x) memset(a,x,sizeof(a))
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double PI = acos(-1.0);
const double eps = 1e-8;
const int maxn = 1e4 + 5;
const int maxs = 20;
char s[maxn];
int sa[maxn], t[maxn], t2[maxn], c[maxn], n , m;
int d[maxn][maxs];
int height[maxn], rank[maxn];
void build_sa(int m)
{
	int i, *x = t, *y = t2;
	for ( i = 0; i < m; i++)		c[i] = 0;
	for ( i = 0; i < n; i++)		c[x[i] = s[i]]++;
	for ( i = 1; i < m; i++)	c[i] += c[i - 1];
	for (i = n - 1; i >= 0; i--)	sa[--c[x[i]]] = i;
	for (int k = 1; k <= n; k <<= 1){
		int p = 0;
		for (i = n - k; i < n; i++)
			y[p++] = i;
		for (i = 0; i < n; i++){
			if (sa[i] >= k)
				y[p++] = sa[i] - k;
		}
		for (i = 0; i < m; i++)	c[i] = 0;
		for (i = 0; i < n; i++)
			c[x[y[i]]]++;
		for (i = 0; i < m; i++)
			c[i] += c[i - 1];
		for (i = n - 1; i >= 0; i--)
			sa[--c[x[y[i]]]] = y[i];
		swap(x, y);
		p = 1;	x[sa[0]] = 0;
		for (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;
	}
}
int cmp_suffix(char *pattern, int p)
{
	return strncmp(pattern, s + sa[p], m);
}
int find(char *P)
{
	m = strlen(P);
	if (cmp_suffix(P, 0) < 0)	return -1;
	if (cmp_suffix(P, n - 1)>0)	return -1;
	int left = 0, right = n - 1 , mid;
	while (left <= right){
		mid = left + (right - left) / 2;
		int res = cmp_suffix(P, mid);
		if (!res)	return mid;
		if (res < 0)		right = mid - 1;
		else                left = mid + 1;
	}
	return -1;
}
void getHeight()
{
	int i, j, k = 0;
	for (i = 0; i < n; i++)	rank[sa[i]] = i;
	for (i = 0; i < n; i++){
		if (k)	k--;
		 j = sa[rank[i] - 1];
		while (s[i + k] == s[j + k])	k++;
		height[rank[i]] = k;
	}
}
void RMQ_init()
{
    for(int i = 0 ; i<n; i++)       d[i][0] = sa[i];
    for(int j = 1 ; (1<<j) - 1 <=n ; j++){
        for(int i = 0 ; i + (1<<j) - 1 < n ; i++){
            d[i][j] = min(d[i][j-1] , d[i + (1<< (j - 1))][ j -1 ]);
        }
    }
}
int RMQ(int L , int R)
{
    int k = 0;
    while((1<<(k + 1)) <= R - L + 1)    k++;
    return min(d[L][k] , d[R - (1 << k) + 1][k]);
}
int main()
{
	char query[1005];
	scanf("%s", s);
	n = strlen(s);
	build_sa('z' + 1 );
	while (scanf("%s", query) == 1){
		int ans = find(query);
		printf("%d\n", ans);
	}
	return 0;
}