bzoj 3238: [Ahoi2013]差异 后缀数组

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=3238

题解 :

先把后缀数组处理出来，之后求出rank[i]与rank[i-1]的最大前缀，最后直接$n^2$暴力搞一搞就好了嘛

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 3010
using namespace std;
int n,p,q,k;
char str[N];
int a[N];
int c[N];
int sa[2][N],rank[2][N];
int h[N];
int f[N][N];
int mx[N];

void calsa(int *sa,int *rank,int *Sa,int *Rank)
{
    for(int i=1;i<=n;i++)c[rank[sa[i]]]=i;
    for(int i=n;i>=1;i--)if(sa[i]>k)Sa[c[rank[sa[i]-k]]--]=sa[i]-k;
    for(int i=n-k+1;i<=n;i++)Sa[c[rank[i]]--]=i;
    for(int i=1;i<=n;i++)
        if(rank[Sa[i-1]]!=rank[Sa[i]]||rank[Sa[i-1]+k]!=rank[Sa[i]+k])
            Rank[Sa[i]]=Rank[Sa[i-1]]+1;
        else Rank[Sa[i]]=Rank[Sa[i-1]];
}

void work()
{
    p=0,q=1,a[0]=-1,a[n+1]=-1;
    for(int i=1;i<=n;i++)c[a[i]]++;
    for(int i=0;i<=2;i++)c[i]+=c[i-1];
    for(int i=1;i<=n;i++)sa[p][c[a[i]]--]=i;
    for(int i=1;i<=n;i++)
        if(a[sa[p][i]]!=a[sa[p][i-1]])
            rank[p][sa[p][i]]=rank[p][sa[p][i-1]]+1;
        else rank[p][sa[p][i]]=rank[p][sa[p][i-1]];
    k=1;
    while(k<n)
    {
        calsa(sa[p],rank[p],sa[q],rank[q]);
        p^=1,q^=1,k<<=1;
    }
}

void getans()
{
    k=0;
    for(int i=1;i<=n;i++)
    {
        if(rank[p][i]==1)h[rank[p][i]]=0;
        else{
            if(k)k--;
            int j=sa[p][rank[p][i]-1];
            while(a[i+k]==a[j+k])k++;
            h[rank[p][i]]=k;
        }
    }
    for(int i=1;i<=n;i++)
    {
        mx[i-1]=h[i];
        for(int j=i-1;j>=1;j--)
            mx[j]=min(mx[j],h[i]);
        for(int j=i-1;j>=1;j--)
        {
            if(!mx[j])break;
            for(int k=mx[j-1]+1;k<=mx[j];k++)
                f[j][k]++;
        }
    }
}

int main()
{
    cin>>n;
    scanf("%s",str+1);
    for(int i=1;i<=n;i++)a[i]=str[i]-'0';
    work();
    getans();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(f[i][j])printf("%d\n",f[i][j]+1);
    return 0;
}