后缀数组板子

重学后缀数组，写完发现常数莫名比以前小了一半，玄学。

/****************************
 * 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;
}

//游说万乘苦不早，著鞭跨马涉远道。
//    -- 李白《南陵别儿童入京》