后缀数组模板

typedef long long ll;
const int maxn = 200010;
int cntA[maxn],cntB[maxn],sa[maxn],tsa[maxn],A[maxn],B[maxn],height[maxn];
int Rank[maxn];
ll n;
char ch[maxn];
void solve()
{
    for(int i = 0;i < 256;i++) cntA[i] = 0;
    for(int i = 1;i <= n;i++) cntA[ch[i-1]]++;
    for(int i = 1;i < 256;i++) cntA[i] += cntA[i-1];
    for(int i = n;i;i--) sa[cntA[ch[i-1]]--] = i;

    Rank[sa[1]] = 1;
    for(int i = 2;i <= n;i++)
    {
        Rank[sa[i]] = Rank[sa[i-1]];
        if(ch[sa[i]-1] != ch[sa[i-1]-1]) Rank[sa[i]]++;
    }

    for(int l = 1;Rank[sa[n]] < n;l <<= 1)
    {
        memset(cntA,0,sizeof(cntA));
        memset(cntB,0,sizeof(cntB));
        for(int i = 1;i <= n;i++)
        {
            cntA[A[i] = Rank[i]]++;
            cntB[B[i] = (i+l <= n)?Rank[i+l]:0]++;
        }
        for(int i = 1;i <= n;i++) cntB[i] += cntB[i-1];
        for(int i = n;i; i--) tsa[cntB[B[i]]--] = i;
        for(int i = 1;i <= n;i++) cntA[i] += cntA[i-1];
        for(int i = n; i;i--) sa[cntA[A[tsa[i]]]--] = tsa[i];

        Rank[sa[1]] = 1;
        for(int i = 2;i <= n;i++)
        {
            Rank[sa[i]] = Rank[sa[i-1]];
            if(A[sa[i]] != A[sa[i-1]] || B[sa[i]] != B[sa[i-1]]) Rank[sa[i]]++;
        }

    }
    for(int i = 1,j = 0;i <= n;i++)
    {
        if(j) j--;
        while(ch[i+j-1] == ch[sa[Rank[i]-1] + j - 1]) j++;
        height[Rank[i]] = j;
    }
}