BZOJ 3238 [Ahoi2013]差异后缀数组+单调栈

最新推荐文章于 2019-12-22 21:09:52 发布

原创最新推荐文章于 2019-12-22 21:09:52 发布 · 2.3k 阅读

2 ·

CC 4.0 BY-SA版权

BZOJ刷题录同时被 2 个专栏收录

225 篇文章

订阅专栏

Suffix Array

6 篇文章

订阅专栏

本文介绍了一种结合后缀数组和单调栈解决字符串匹配问题的方法。通过详细解析算法步骤，展示如何高效地求解特定字符串操作问题。

题意:链接

方法:后缀数组+单调栈

解析:

一眼题，刷水有益健康23333

首先一看这个式子就想能不能O(1)…这是病

然后发现不能O(1),但是可以转化？

好吧就先看前面的len求和吧。

假设串长为n，则原式中不看lcp，len的求和为 $(n-1)*\sum\limits_{i = 1}^n i$ ;

然后我们再来看减去的lcp部分。

首先，讲后缀数组的时候wyf大爷讲到了如何利用height知道任意两个后缀串的lcp，即从二者height下标构成的区间中的height的最小值。

所以减去的部分转化到了给定序列，求所有区间的最小值的和乘2。

先看范围O(n^2)枚举被排除。QAQ

然后想优化呗..数据结构什么的都路过了一下

但是…MD这不就是某次考试题么- -!单调栈求某个点向左能扩展到哪，向右能扩展到哪。两边扩展长度的乘积不就是以该点的值为最小值的区间个数么..

O(nlogn+n)完美水过(为什么我自带常数)

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 500010
using namespace std;
typedef long long ll;
ll n;
char s[N];
int sa[N],rnk[N],cnt[N],t0[N],t1[N];
ll height[N];
void getsa(int m)
{
    int *x=t0,*y=t1;
    for(int i=0;i<m;i++)cnt[i]=0;
    for(int i=0;i<n;i++)cnt[x[i]=s[i]]++;
    for(int i=1;i<m;i++)cnt[i]+=cnt[i-1];
    for(int i=n-1;i>=0;i--)sa[--cnt[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k;i<n;i++)y[p++]=i;
        for(int i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
        for(int i=0;i<m;i++)cnt[i]=0;
        for(int i=0;i<n;i++)cnt[x[y[i]]]++;
        for(int i=1;i<m;i++)cnt[i]+=cnt[i-1];
        for(int i=n-1;i>=0;i--)sa[--cnt[x[y[i]]]]=y[i];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&sa[i]+k<n&&sa[i-1]+k<n&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p==n)break;
        m=p;
    }
}
void getheight()
{
    int k=0;
    for(int i=0;i<n;i++)rnk[sa[i]]=i;
    for(int i=0;i<n;i++)
    {
        if(k!=0)k--;
        if(!rnk[i])continue;
        int j=sa[rnk[i]-1];
        while(s[i+k]==s[j+k])k++;
        height[rnk[i]]=k;
    }
}
ll ans;
int sta[N];
int top;
ll l[N],r[N];
int main()
{
    scanf("%s",s);
    n=strlen(s);
    ans+=(n-1)*(1+n)*n/2;
    getsa(256);
    getheight();
    n--;
    for(int i=1;i<=n;i++)
    {
        while(top&&height[i]<height[sta[top]])r[sta[top--]]=i-1;
        sta[++top]=i;
    }
    while(top)r[sta[top--]]=n;
    for(int i=n;i>=1;i--)
    {
        while(top&&height[i]<=height[sta[top]])l[sta[top--]]=i+1;
        sta[++top]=i;
    }
    while(top)l[sta[top--]]=1;
    for(ll i=1;i<=n;i++)
    {
        ans-=2ll*height[i]*(i-l[i]+1)*(r[i]-i+1); 
    }
    printf("%lld\n",ans);
}