bzoj 4516: [Sdoi2016]生成魔咒 后缀数组

题意

给你一个字符串，问该字符串的每一个前缀中分别有多少个不同的子串。
n<=100000

分析

一种很暴力的方法就是构建SAM，然后在更新fa的时候暴力维护答案。

我用的是sa来做。我们考虑将字符串反过来，构建sa，那么一个原串的一个前缀就变成了当前的一个后缀。
设当前处理到i，1到i-1已经处理完毕，且答案为ans，那么我们只要统计原串中以i结尾的后缀与以j(1<=j<=i-1)结尾的后缀的最长公共前缀是多少即可。字符串倒置后就变成了求以i开头的后缀与其排名最近且编号小于i的最长公共前缀，这个可以用一个set来维护，然后统计答案即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<set>
#define N 100005
#define LL long long
using namespace std;

int n,s[N],b[N],c[N],d[N],sa[N],rank[N*2],height[N],rmq[N][25],a[N];
set<int> w;

void get_sa(int n)
{
    for (int i=1;i<=n;i++) b[s[i]]++;
    for (int i=1;i<=n;i++) b[i]+=b[i-1];
    for (int i=n;i>=1;i--) c[b[s[i]]--]=i;
    int t=0;
    for (int i=1;i<=n;i++)
    {
        if (s[c[i]]!=s[c[i-1]]) t++;
        rank[c[i]]=t;
    }
    int j=1;
    while (j<=n)
    {
        for (int i=1;i<=n;i++) b[i]=0;
        for (int i=1;i<=n;i++) b[rank[i+j]]++;
        for (int i=1;i<=n;i++) b[i]+=b[i-1];
        for (int i=n;i>=1;i--) c[b[rank[i+j]]--]=i;
        for (int i=1;i<=n;i++) b[i]=0;
        for (int i=1;i<=n;i++) b[rank[i]]++;
        for (int i=1;i<=n;i++) b[i]+=b[i-1];
        for (int i=n;i>=1;i--) d[b[rank[c[i]]]--]=c[i];
        t=0;
        for (int i=1;i<=n;i++)
        {
            if (rank[d[i]]!=rank[d[i-1]]||rank[d[i]]==rank[d[i-1]]&&rank[d[i]+j]!=rank[d[i-1]+j]) t++;
            c[d[i]]=t;
        }
        for (int i=1;i<=n;i++) rank[i]=c[i];
        if (t==n) break;
        j*=2;
    }
    for (int i=1;i<=n;i++) sa[rank[i]]=i;
}

void get_height(int n)
{
    int k=0;
    for (int i=1;i<=n;i++)
    {
        if (k) k--;
        int j=sa[rank[i]-1];
        while (j+k<=n&&i+k<=n&&s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
}

void get_rmq(int n)
{
    for (int i=1;i<=n;i++) rmq[i][0]=height[i];
    int lg=log(n)/log(2);
    for (int j=1;j<=lg;j++)
        for (int i=1;i<=n-(1<<j)+1;i++)
            rmq[i][j]=min(rmq[i][j-1],rmq[i+(1<<(j-1))][j-1]);
}

int get_min(int l,int r)
{
    int lg=log(r-l+1)/log(2);
    return min(rmq[l][lg],rmq[r-(1<<lg)+1][lg]);
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&s[n-i+1]);
    for (int i=1;i<=n;i++) a[i]=s[i];
    sort(a+1,a+n+1);
    int a1=unique(a+1,a+n+1)-a-1;
    for (int i=1;i<=n;i++) s[i]=lower_bound(a+1,a+a1+1,s[i])-a;
    get_sa(n);
    get_height(n);
    get_rmq(n);
    w.insert(rank[n]);
    LL ans=1;
    printf("%lld\n",ans);
    for (int i=n-1;i>=1;i--)
    {
        int mx=0;
        set<int>::iterator it=w.upper_bound(rank[i]);
        if (it!=w.end()) mx=max(mx,get_min(rank[i]+1,*it));
        if (it!=w.begin()) it--,mx=max(mx,get_min(*it+1,rank[i]));
        w.insert(rank[i]);
        ans+=n-i+1-mx;
        printf("%lld\n",ans);
    }
    return 0;
}