【SDOI2016】bzoj4516 生成魔咒【解法二】

在后面添加字符的操作很不舒服，可以倒过来看成是添加后缀。每加入一个后缀，在统计的时候需要去掉之前已经出现过的前缀。在SA上面找前后最近的已经被加入的位置求LCP即可。找最近的位置可以用set维护。

#include<cstdio>
#include<cstring>
#include<set>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=200010;
set<int> s;
int sa[maxn],rank[maxn],f[maxn],cnt[maxn],a[maxn],ord[maxn],height[maxn],
mn[maxn][20],log[maxn],n,m;
int qry(int l,int r)
{
    int k=log[r-l+1];
    return min(mn[l][k],mn[r-(1<<k)+1][k]);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int p,now,x;
    LL ans=0;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&a[n-i+1]);
        ord[i]=a[n-i+1];
    }
    sort(ord+1,ord+n+1);
    m=unique(ord+1,ord+n+1)-ord-1;
    for (int i=1;i<=n;i++)
        f[i]=lower_bound(ord+1,ord+m+1,a[i])-ord;
    for (int i=1;i<=n;i++) cnt[rank[i]=f[i]]++;
    for (int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
    for (int i=n;i;i--) sa[cnt[f[i]]--]=i;
    for (int k=1;;k++)
    {
        p=0;
        for (int i=n-(1<<k-1)+1;i<=n;i++) f[++p]=i;
        for (int i=1;i<=n;i++)
            if (sa[i]-(1<<k-1)>0)
                f[++p]=sa[i]-(1<<k-1);
        for (int i=1;i<=m;i++) cnt[i]=0;
        for (int i=1;i<=n;i++) cnt[rank[f[i]]]++;
        for (int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
        for (int i=n;i;i--) sa[cnt[rank[f[i]]]--]=f[i];
        for (int i=1;i<=n;i++) f[i]=rank[i];
        rank[sa[1]]=1;
        for (int i=2;i<=n;i++)
            if (f[sa[i]]!=f[sa[i-1]]||f[sa[i]+(1<<k-1)]!=f[sa[i-1]+(1<<k-1)])
                rank[sa[i]]=rank[sa[i-1]]+1;
            else
                rank[sa[i]]=rank[sa[i-1]];
        m=rank[sa[n]];
        if (m>=n) break;
    }
    for (int i=1;i<=n;i++)
    {
        height[rank[i]]=height[rank[i-1]];
        if (height[rank[i]]) height[rank[i]]--;
        while (a[i+height[rank[i]]]==a[sa[rank[i]-1]+height[rank[i]]])
            height[rank[i]]++;
    }
    for (int k=1;(1<<k)<=n;k++) log[1<<k]=k;
    for (int i=3;i<=n;i++)
        if (!log[i]) log[i]=log[i-1];
    for (int i=1;i<=n;i++) mn[i][0]=height[i];
    for (int k=1;k<=log[n];k++)
        for (int i=1;i+(1<<k)-1<=n;i++)
            mn[i][k]=min(mn[i][k-1],mn[i+(1<<k-1)][k-1]);
    s.insert(0);
    s.insert(n+1);
    for (int i=n;i;i--)
    {
        now=0;
        x=*(--s.lower_bound(rank[i]));
        if (x) now=max(now,qry(x+1,rank[i]));
        x=*(s.lower_bound(rank[i]));
        if (x<=n) now=max(now,qry(rank[i]+1,x));
        ans+=n-i+1-now;
        s.insert(rank[i]);
        printf("%lld\n",ans);
    }
}