重学后缀数组,写完发现常数莫名比以前小了一半,玄学。
/****************************
* Au: Hany01
* Prob: Suffix Array
* Date: Jan 30th, 2018
* Email: hany01@foxmail.com
****************************/
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
typedef vector<int> VI;
#define rep(i , j) for (int i = 0 , i##_end_ = j; i < i##_end_ ; ++ i)
#define For(i , j , k) for (int i = (j) , i##_end_ = (k) ; i <= i##_end_ ; ++ i)
#define Fordown(i , j , k) for (int i = (j) , i##_end_ = (k) ; i >= i##_end_ ; -- i)
#define Set(a , b) memset(a , b , sizeof(a))
#define SZ(a) ((int)(a.size()))
#define ALL(a) a.begin(), a.end()
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define Mod (1000000007)
#define y1 wozenmezhemecaia
#ifdef hany01
#define debug(...) fprintf(stderr , __VA_ARGS__)
#else
#define debug(...)
#endif
inline void File() {
#ifdef hany01
freopen("sa.in" , "r" , stdin);
freopen("sa.out" , "w" , stdout);
#endif
}
template<typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template<typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }
char c_; int _ , __;
inline int read() {
for (_ = 0 , __ = 1 , c_ = getchar() ; !isdigit(c_) ; c_ = getchar()) if (c_ == '-') __ = -1;
for ( ; isdigit(c_) ; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
return _ * __;
}
const int maxn = 100005;
int n, m, c[maxn], sa[maxn], tp[maxn], rk[maxn], height[maxn];
char s[maxn];
//c[i]: 桶
//sa[i]: 第i名的位置
//rk[i]: 以i开头的后缀的排名
//tp[i]: 第二关键字排名为i的位置
//height[i]: 排名第i个和第i-1个的LCP
inline void Radix_Sort()//基数排序
{
For(i, 1, m) c[i] = 0;
For(i, 1, n) ++ c[rk[i]];
For(i, 1, m) c[i] += c[i - 1];
Fordown(i, n, 1) sa[c[rk[tp[i]]] --] = tp[i];
}
inline void Build_SA()
{
For(i, 1, n) rk[i] = s[i], tp[i] = i;
m = 255; Radix_Sort();
for (register int k = 1, p; k <= n; k <<= 1) {
p = 0;
For(i, n - k + 1, n) tp[++ p] = i;//第二关键字为空的排在最前面
For(i, 1, n) if (sa[i] > k) tp[++ p] = sa[i] - k;//依次根据上次的计算结果算出第二关键字的sa
Radix_Sort(); swap(rk, tp);//将上次的排名存在tp中
rk[sa[1]] = 1, m = 1;//m表示不同的字符的个数
For(i, 2, n) rk[sa[i]] = tp[sa[i]] == tp[sa[i - 1]] && tp[sa[i] + k] == tp[sa[i - 1] + k] ? m : ++ m;
if (m >= n) return ;//如果每一个后缀都得到了一个不同的排名,那么直接终止
}
}
inline void Build_Height()
{
//利用性质:height[rank[i]] >= height[rank[i - 1]] - 1
for (register int i = 1, j, k = 0; i <= n; height[rk[i ++]] = k)
for (k = k ? k - 1 : 0, j = sa[rk[i] - 1]; s[j + k] == s[i + k]; ++ k) ;
}
int main()
{
File();
scanf("%s", s + 1), n = strlen(s + 1);
Build_SA();
For(i, 1, n) printf("%d ", sa[i]); putchar('\n');
Build_Height();
For(i, 2, n) printf("%d ", height[i]); putchar('\n');
return 0;
}
//游说万乘苦不早,著鞭跨马涉远道。
// -- 李白《南陵别儿童入京》