题目大意:给出一个字符串,循环字符串的起始位置,可以得到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;
}