後綴數組模板

後綴數組的模板,這樣說明就很詳細了吧!


/*
 *	後綴數組模板-倍增法
 *	使用方法:
 *		1、讀取字符串轉換成int數組,長度爲len,下標從0開始
 *		2、在字符串末尾加一字典序最小字符,一般爲0,並找到最大的字符設爲maxa
 *		3、調用函數da(num,sa,len+1,maxa+1)
 *				求得的sa數組的含義: sa[i]爲第i字典序後綴字符串的首字母下標
 *		4、調用函數build_height(num,sa,len)
 *				求得的ranking數組的含義: ranking[i]爲首字母下標爲i的後綴數組的字典序的名次
 *				求得的height數組含義: height[i]爲第i字典序後綴與第i-1字典序後綴的最長公共前綴,height[1] = 0;
 *				h[i]=height[rank[i]],也就是suffix(i)和在它前一名的後綴的最長公共前綴。
 *	注意事項:
 *		轉化爲數字的字符串最小數值必須不小於1,如果是0的話會Runtime Error,請自重.......
 */
#include <iostream>
#include <algorithm>
#include <cstdio>
#define INF 2*0x3f3f3f3f

using namespace std;
const int maxn = 200001; //注意數組的大小,記得更改

int wa[maxn], wb[maxn], wv[maxn], wss[maxn];
int cmp(int *r, int a, int b, int l) {
	return r[a] == r[b] && r[a + l] == r[b + l];
}
void da(int *r, int *sa, int n, int m) { //r爲所求取的轉化字符串 sa爲求得的sa數組  n爲數組長度+1  m爲數組的最大值+1
	int i, j, p, *x = wa, *y = wb, *t;
	for (i = 0; i < m; i++) wss[i] = 0;
	for (i = 0; i < n; i++) wss[x[i] = r[i]]++;
	for (i = 1; i < m; i++) wss[i] += wss[i - 1];
	for (i = n - 1; i >= 0; i--) sa[--wss[x[i]]] = i;
	for (j = 1, p = 1; p < n; j *= 2, m = p) {
		for (p = 0, i = n - j; i < n; i++) y[p++] = i;
		for (i = 0; i < n; i++) if (sa[i] >= j) y[p++] = sa[i] - j;
		for (i = 0; i < n; i++) wv[i] = x[y[i]];
		for (i = 0; i < m; i++) wss[i] = 0;
		for (i = 0; i < n; i++) wss[wv[i]]++;
		for (i = 1; i < m; i++) wss[i] += wss[i - 1];
		for (i = n - 1; i >= 0; i--) sa[--wss[wv[i]]] = y[i];
		for (t = x, x = y, y = t, p = 1, x[sa[0]] = 0, i = 1; i < n; i++)
			x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1 : p++;
	}
	return;
}

int ranking[maxn], height[maxn];
void build_height(int *r, int *sa, int n) { //r爲求得的轉化原始數組,sa爲已經求得的數組,n爲len
	int i, j, k = 0;
	for (i = 1; i <= n; i++) ranking[sa[i]] = i;
	for (i = 0; i < n; height[ranking[i++]] = k) {
		for (k ? k-- : 0, j = sa[ranking[i] - 1]; r[i + k] == r[j + k]; k++);
	}
}

int sa[maxn], r[maxn];

int main() {
	freopen("in.in", "r", stdin);
	freopen("out.out", "w", stdout);
	return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章