首页 > 代码库 > BZOJ 1031 JSOI 2007 字符加密Cipher 后缀数组

BZOJ 1031 JSOI 2007 字符加密Cipher 后缀数组

题目大意:给出一个字符串,循环字符串的起始位置,可以得到length个不同的字符串,问把这些字符串排序之后每一个字符串的第一位是什么。


思路:后缀数组裸题,只需要将整个字符串倍增,然后求一次sa。


CODE:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAX 200010
using namespace std;

char s[MAX];
int len;

int sa[MAX],val[MAX];

inline bool Same(int x,int y,int l,int len)
{
	return val[x] == val[y] &&
	((x + l >= len && y + l >= len) || (x + l < len && y + l < len && val[x + l] == val[y + l]));
}

void GetSaffixArray()
{
	static int cnt[MAX],_val[MAX],q[MAX],lim = 256,j;
	for(int i = 0; i < lim; ++i)		cnt[i] = 0;
	for(int i = 0; i < len; ++i)		++cnt[val[i] = s[i]];
	for(int i = 1; i < lim; ++i)		cnt[i] += cnt[i - 1];
	for(int i = len - 1; i >= 0; --i)	sa[--cnt[val[i]]] = i;
	for(int d = 1;; ++d) {
		int top = 0,l = 1 << (d - 1);
		for(int i = 0; i < len; ++i)	if(sa[i] + l >= len)	q[top++] = sa[i];
		for(int i = 0; i < len; ++i)	if(sa[i] >= l)	q[top++] = sa[i] - l;
		
		for(int i = 0; i < lim; ++i)		cnt[i] = 0;
		for(int i = 0; i < len; ++i)		++cnt[val[q[i]]];
		for(int i = 1; i < lim; ++i)		cnt[i] += cnt[i - 1];
		for(int i = len - 1; i >= 0; --i)	sa[--cnt[val[q[i]]]] = q[i];
		lim = 0;
		for(int i = 0; i < len; ++lim) {
			for(j = i; j < len - 1 && Same(sa[j],sa[j + 1],l,len); ++j);
			for(; i <= j; ++i)	_val[sa[i]] = lim;
		}
		for(int i = 0; i < len; ++i)	val[i] = _val[i];
		if(lim == len)	break;
	}
	return ;
}

int main()
{
	scanf("%s",s);
	len = strlen(s);
	for(int i = len; i < len << 1; ++i)
		s[i] = s[i - len];
	s[len << 1] = '\0';
	len = strlen(s);
	GetSaffixArray();
	for(int i = 0; i < len; ++i)
		if(sa[i] < (len >> 1))
			putchar(s[sa[i] + (len >> 1) - 1]);
	return 0;
}


BZOJ 1031 JSOI 2007 字符加密Cipher 后缀数组