后缀数组模板

过了期末了，继续写ACM题

自己写的后缀数组模板。

k, len, rank[], sa[], tmp[], 都要写在外面，这样就不用来回折腾了。

也是使用倍增法来做的，得到一个sa，保存了后缀排在第i位的后缀的起始位置。

因为空也算一个后缀，所以函数中都是 <= len

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAX 2000
#define MS(x) memset(x, 0, sizeof(x))
char s[MAX];
int sa[MAX];
int rank[MAX];
int tmp[MAX];
int len;
int k;
bool compare_sa(int i, int j)
{
	if(rank[i] != rank[j])
		return rank[i] < rank[j];
	else
	{
		int ri = i + k <= len ? rank[i + k] : -1;
		int rj = j + k <= len ? rank[j + k] : -1;
		return ri < rj;
	}
}

void construct_sa()
{
	MS(sa);
	MS(rank);
	MS(tmp);
	len = strlen(s);
	for(int i = 0; i <= len; i++)
	{
		sa[i] = i;
		rank[i] = i < len ? s[i] : -1;
	}
	for(k = 1; k <= len; k *= 2)
	{
		sort(sa, sa + len + 1, compare_sa);
		tmp[sa[0]] = 0;
		for(int i = 1; i <= len; i++)
		{
			tmp[sa[i]] = tmp[sa[i-1]] + ((compare_sa(sa[i - 1], sa[i])) ? 1 : 0);
		}
		for(int i = 0; i <= len; i++)
			rank[i] = tmp[i];
	}

}

int main()
{
	scanf("%s", s);
	len=strlen(s);
	construct_sa();
	for(int i = 0; i <= len; i++)
		printf("%d ", sa[i]);
	puts("\n");
}