BZOJ 2251

本题模版题就不提示了。

我用的后缀数组， 网上也有用 $Trie$ ， 因为此题的输出本来就是 $n^2$ 级别的，所以理论最优时间复杂度就是 $n^2$

看到很多小伙伴建完后缀数组后就ST标 + 二分暴力。 这样做是 $n^2 \log(n)$ 的（此题只要做法正确怎么都能过？） 而且有些细节需要处理。我这里提供一种直接递推处理的方法。

我们现在按照后缀排序的顺序逐个考虑每个后缀 $Suf$ , $Suf$ 中长度 $\le 和前面的最长公共前缀$ 的部分显然应该在前面就输出了， 我们不妨从后往前递推，这样处理相当简洁明了，可以参考代码最后的递推。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3100;

int n;
char s[maxn];

int a[maxn] , sa[maxn] , t1[maxn] , t2[maxn] , c[maxn] , Rank[maxn] , height[maxn];

void SA(int m)
{
    int *x = t1 , *y = t2;

    for(int i=0;i<m;i++) c[i] = 0;
    for(int i=0;i<n;i++) c[x[i] = a[i]]++;
    for(int i=1;i<m;i++) c[i] += c[i-1];
    for(int i=0;i<n;i++) sa[--c[x[i]]] = i;

    for(int k=1;;k<<=1)
    {
        int p = 0;
        for(int i=n-k;i<n;i++) y[p++] = i;
        for(int i=0;i<n;i++) if(sa[i] >= k) y[p++] = sa[i] - k;

        for(int i=0;i<m;i++) c[i] = 0;
        for(int i=0;i<n;i++) c[x[i]]++;
        for(int i=1;i<m;i++) c[i] += c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]] = y[i];

        swap(x, y);
        p = 0;
        x[sa[0]] = p++;
        for(int i=1;i<n;i++) x[sa[i]] = y[sa[i]] == y[sa[i-1]] && y[sa[i]+k] == y[sa[i-1]+k] ? p-1 : p++;
        if((m = p) >= n) break;
    }
}

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

int res[maxn][maxn];

int main(int argc, char *argv[]) {

    cin>>n;
    scanf("%s" , s);
    for(int i=0;i<n;i++) a[i] = s[i] - '0' + 1;

    ++n;
    SA(3);
    getHeight();
//  for(int i=0;i<n;i++) cout<<height[i]<<" "; cout<<endl;
    for(int i=1;i<n;i++) for(int j=1;j<=n-sa[i];j++) res[i][j]++;
    for(int i=n-1;i;i--) for(int j=1;j<=height[i];j++) res[i-1][j]+=res[i][j];

    for(int i=1;i<n-1;i++) for(int j=height[i]+1;j<=height[i+1];j++) printf("%d\n" , res[i][j]);

    return 0;
}