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