bzoj4516:[Sdoi2016]生成魔咒

本文深入探讨了字符串处理中的一种高效算法——SA算法,详细解释了如何通过该算法求解字符串的后缀数组,以及如何利用后缀数组解决实际问题。文章通过具体的代码示例,展示了算法的具体实现过程,包括如何构建后缀数组、如何求解最长公共前缀等问题。此外,还介绍了如何在实际应用中优化算法效率。

传送门

这个题目和我的另一篇博客做法很像
做完那个几乎就会这个了吧
如果你不会,你可以继续往下看
实际上由于当前的数的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]);
}

转载于:https://www.cnblogs.com/lcxer/p/10460893.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值