BZOJ4566: [Haoi2016]找相同字符

BZOJ4566: [Haoi2016]找相同字符

Description

给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。
两个方案不同当且仅当这两个子串中有一个位置不同。

Input

两行,两个字符串s1,s2,长度分别为n1,n2。

1 <=n1, n2<= 200000,字符串中只有小写字母

Output

输出一个整数表示答案

Sample Input

aabb
bbaa

Sample Output

10

题解Here!

​$O(n^3)$的大暴力:

首先,我们知道。如果连个串中各取一个极长的相同的子串(极长的意思就是这两个串再向两边延伸就会不同或出界),则这个极长的串的所有子串也是相同的。

由于枚举极长的串并求出其长度并不方便,所以我们可以枚举这个极长的串的左端点,再求出其长度。

这样就可以不重不漏找出所有的极长的串。

更图方便的话,我们可以只考虑两原串中某后缀的所有前缀,这可以补充不漏找出所有的子串。

但是由于我们要在两个字符串中枚举,同时找出子串的长度也是$O(n)$的,于是总复杂度为$O(n^3)$。

$O(n^2)$的优化大暴力:

​刚才说道了后缀的前缀,我们不由自主想到了后缀数组。

如果我们可以很快求出$A$串和$B$串的某两个后缀的最长公共前缀,我们就可以将时间复杂度优化的更低。

这不就是$height$数组解决的嘛!

所以我们可以将两个字符串通过一个分隔符拼接起来,求出$height$数组,即按字典序排序后每个后缀和前一个的$LCP$。

再利用$height$数组和$ST$表,我们就可以$O(1)$得到任意两后缀的$LCP$。

因此,这样做的时间复杂度只有枚举子串起点的复杂度了,即$O(n^2)$。

$O(nlog_2n)$的正解:

​现在拉高时间复杂度的罪魁祸首就是枚举起点了。

所以我们想将其复杂度降低。

考虑到利用$height$数组求任意两个后缀的$LCP$时的独特性质:

两个后缀的$LCP$为字典序排序后他们中间夹的最小的$height$。

也就是说排序后,一个后缀越往后数$LCP$的长度越小。

这样,我们就可以用单调栈维护这个最小值。

分$A$串的子串在前、$B$的子串在前两种情况分别用单调栈求出答案,加起来就行。

至于如何维护,emmm......

​ 由于利用了单调栈这个神奇的手段,时间复杂度降到了$O(nlog_2n)$,也就是倍增处理后缀数组的复杂度。

当然你可以用$DC3$来完成,复杂度可以降到$O(n)$。(当然,本蒟蒻不会啦。。。)

附代码:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define MAXN 400010
using namespace std;
pair<int,long long> stack[MAXN];
int n,l1,l2;
int val[MAXN],sum[MAXN];
char str[MAXN];
int top,sa[MAXN],rk[MAXN],tax[MAXN],tp[MAXN],height[MAXN];
inline int read(){
    int date=0,w=1;char c=0;
    while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
    while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
    return date*w;
}
void radixsort(){
    for(int i=0;i<=top;i++)tax[i]=0;
    for(int i=1;i<=n;i++)tax[rk[i]]++;
    for(int i=1;i<=top;i++)tax[i]+=tax[i-1];
    for(int i=n;i>=1;i--)sa[tax[rk[tp[i]]]--]=tp[i];
}
void suffixsort(){
    top=30;
    for(int i=1;i<=n;i++){
        rk[i]=val[i];
        tp[i]=i;
    }
    radixsort();
    for(int w=1,p=0;p<n;top=p,w<<=1){
        p=0;
        for(int i=1;i<=w;i++)tp[++p]=n-w+i;
        for(int i=1;i<=n;i++)if(sa[i]>w)tp[++p]=sa[i]-w;
        radixsort();
        swap(tp,rk);
        rk[sa[1]]=p=1;
        for(int i=2;i<=n;i++)
        rk[sa[i]]=(tp[sa[i-1]]==tp[sa[i]]&&tp[sa[i-1]+w]==tp[sa[i]+w])?p:++p;
    }
}
void getheight(){
    for(int i=1,j,k=0;i<=n;i++){
        if(k)k--;
        j=sa[rk[i]-1];
        while(val[i+k]==val[j+k])k++;
        height[rk[i]]=k;
    }
}
void work(){
    long long ans=0;
    top=0;
    stack[0]=make_pair(1,0);
    for(int i=1;i<=n;i++)sum[i]=sum[i-1]+(sa[i]<=l1);
    for(int i=1;i<=n;i++){
        while(top&&height[i]<height[stack[top].first])top--;
        top++;
        stack[top]=make_pair(i,(long long)(sum[i-1]-sum[stack[top-1].first-1])*height[i]+stack[top-1].second);
        if(sa[i]>l1+1)ans+=stack[top].second;
    }
    top=0;
    for(int i=1;i<=n;i++)sum[i]=sum[i-1]+(sa[i]>l1+1);
    for(int i=1;i<=n;i++){
        while(top&&height[i]<height[stack[top].first])top--;
        top++;
        stack[top]=make_pair(i,(long long)(sum[i-1]-sum[stack[top-1].first-1])*height[i]+stack[top-1].second);
        if(sa[i]<=l1+1)ans+=stack[top].second;
    }
    printf("%lld
",ans);
}
void init(){
    scanf("%s",str+1);
    l1=n=strlen(str+1);
    str[++n]='z'+1;
    scanf("%s",str+n+1);
    n=strlen(str+1);
    for(int i=1;i<=n;i++)val[i]=str[i]-'a'+1;
    suffixsort();
    getheight();
}
int main(){
    init();
    work();
    return 0;
}

以上是后缀数组解法,当然你可以用$SAM$吊打$SA$,就像$AC$自动机吊打$Trie$和$KMP$一样。。。

(坑,未填。。。)

原文地址:https://www.cnblogs.com/Yangrui-Blog/p/9451807.html