bzoj4516:[Sdoi2016]生成魔咒

传送门

这个题目和我的另一篇博客做法很像
做完那个几乎就会这个了吧
如果你不会，你可以继续往下看
实际上由于当前的数的sa不一定连续
所以需要求出任意两个的最长公共前缀
假设\(x,y,y>x\)，就是\(min\{height[x+1],...,height[y]\}\)
然后把原本的断开，把当前数加进去，操作操作就没了
代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=1e5+10;map<int,int>mp;long long ans[maxn];set<int>q;
set<int>::iterator it,itt;
int n,c[maxn],w[maxn],num,tot,a[maxn],x[maxn],m,y[maxn],sa[maxn],rk[maxn],h[maxn],tmp,g[maxn];
struct oo{int l,r,mn;}s[maxn*4];
void update(int x){s[x].mn=min(s[x<<1].mn,s[x<<1|1].mn);}
void build(int x,int l,int r)
{
	s[x].l=l,s[x].r=r;int mid=(l+r)>>1;
	if(l==r){s[x].mn=h[l];return ;}
	build(x<<1,l,mid),build(x<<1|1,mid+1,r);
	update(x);
}
int get(int x,int l,int r)
{
	if(l>r)return 0;
	if(l<=s[x].l&&r>=s[x].r)return s[x].mn;
	int mid=(s[x].l+s[x].r)>>1,ans=1e9;
	if(l<=mid)ans=min(ans,get(x<<1,l,r));
	if(r>mid)ans=min(ans,get(x<<1|1,l,r));
	return ans;
}
int main()
{
    read(n);
    for(rg int i=1;i<=n;i++)read(c[i]),w[i]=c[i];sort(w+1,w+n+1);
    for(rg int i=1;i<=n/2;i++)swap(c[i],c[n-i+1]);
    for(rg int i=1;i<=n;i++)if(w[i]!=w[i-1])mp[w[i]]=++tot;m=tot;
    for(rg int i=1;i<=n;i++)a[x[i]=mp[c[i]]]++;
    for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
    for(rg int i=n;i;i--)sa[a[x[i]]--]=i;
    for(rg int k=1;k<=n;k<<=1,num=0)
    {
        for(rg int i=n-k+1;i<=n;i++)y[++num]=i;
        for(rg int i=1;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
        for(rg int i=1;i<=m;i++)a[i]=0;
        for(rg int i=1;i<=n;i++)a[x[i]]++;
        for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
        for(rg int i=n;i;i--)sa[a[x[y[i]]]--]=y[i];
        for(rg int i=1;i<=n;i++)y[i]=x[i];
        num=x[sa[1]]=1;
        for(rg int i=2;i<=n;i++)
            if(y[sa[i]]!=y[sa[i-1]]||y[sa[i]+k]!=y[sa[i-1]+k])x[sa[i]]=++num;
            else x[sa[i]]=num;
        if(num>=n)break;m=num;
    }
    for(rg int i=1;i<=n;i++)rk[sa[i]]=i;
    for(rg int i=1,k=0,j;i<=n;h[rk[i++]]=k)
        for(k=k?k-1:k,j=sa[rk[i]-1];c[j+k]==c[i+k];k++);
    build(1,1,n);
    for(rg int i=n;i;i--)
	{
		ans[i]=ans[i+1]+n-i+1;
		q.insert(rk[i]);
		it=itt=q.lower_bound(rk[i]);itt++;
		if(it!=q.begin()){it--;tmp=get(1,(*it)+1,rk[i]);ans[i]+=g[*it],ans[i]-=tmp,g[*it]=tmp;}
		if(itt!=q.end()){g[rk[i]]=get(1,rk[i]+1,*itt),ans[i]-=g[rk[i]];}
	}
    for(rg int i=n;i;i--)printf("%lld\n",ans[i]);
}