BZOJ 2251

本题模版题就不提示了。

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

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

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

#include 
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;i0;
    for(int i=0;ifor(int i=1;i1];
    for(int i=0;ifor(int k=1;;k<<=1)
    {
        int p = 0;
        for(int i=n-k;ifor(int i=0;iif(sa[i] >= k) y[p++] = sa[i] - k;

        for(int i=0;i0;
        for(int i=0;ifor(int i=1;i1];
        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;i1]] && y[sa[i]+k] == y[sa[i-1]+k] ? p-1 : p++;
        if((m = p) >= n) break;
    }
}

void getHeight()
{
    for(int i=0;ifor(int i=0,k=0,j;i1;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'0' + 1;

    ++n;
    SA(3);
    getHeight();
//  for(int i=0;i
    for(int i=1;ifor(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;i1;i++) for(int j=height[i]+1;j<=height[i+1];j++) printf("%d\n" , res[i][j]);

    return 0;
}

你可能感兴趣的:(后缀数组,递推)