[BZOJ4566][HAOI2016]找相同字符(SAM)

4566: [Haoi2016]找相同字符

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 950  Solved: 550
[Submit][Status][Discuss]

Description

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

个子串中有一个位置不同。

Input

两行，两个字符串s1，s2，长度分别为n1，n2。1 <=n1, n2<= 200000，字符串中只有小写字母

Output

输出一个整数表示答案

Sample Input

aabb
bbaa

Sample Output

10

HINT

Source

[Submit][Status][Discuss]

后缀数组+单调栈可以做，但是后缀数组的超长模板加上单调栈的超多细节，想想就复杂。

对于热爱DP或码力不够的选手，SAM则是最佳选择。

给两个式子自己体会什么意思，很好理解。（tmp表示当前匹配上的长度）

sum[i]=sum[fa[i]]+(l[i]-l[fa[i]])*right[i],ans+=sum[fa[p]]+(tmp-l[fa[p]])*right[p]

radixsort然后拓扑DP即可，模板又打错了。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define rep(i,l,r) for (int i=l; i<=r; i++)
typedef long long ll;
using namespace std;

const int N=400100;
char s1[N],s2[N];
int n,m,p,lst=1,np,cnt=1,mx[N],right[N],fa[N],son[N][26],q[N],c[N];
ll ans,sum[N];

void ext(int c){
    p=lst; lst=np=++cnt; mx[np]=mx[p]+1; right[np]++;
    while (!son[p][c] && p) son[p][c]=np,p=fa[p];
    if (!p) fa[np]=1;
    else{
        int q=son[p][c];
        if (mx[q]==mx[p]+1) fa[np]=q;
        else{
            int nq=++cnt; mx[nq]=mx[p]+1;
            memcpy(son[nq],son[q],sizeof(son[q]));
            while (p && son[p][c]==q) son[p][c]=nq,p=fa[p];
            fa[nq]=fa[q]; fa[q]=fa[np]=nq;
        }
    }
}

void radix(){
    rep(i,0,n) c[i]=0;
    rep(i,1,cnt) c[mx[i]]++;
    rep(i,1,n) c[i]+=c[i-1];
    for (int i=cnt; i; i--) q[c[mx[i]]--]=i;
    for (int i=cnt; i; i--) right[fa[q[i]]]+=right[q[i]];
    rep(i,1,cnt) sum[q[i]]=sum[fa[q[i]]]+(mx[q[i]]-mx[fa[q[i]]])*right[q[i]];
}

void solve(){
    int tmp=0; p=1;
    rep(i,1,m){
        int c=s2[i]-'a';
        if (son[p][c]) p=son[p][c],tmp++;
        else{
            while (p && !son[p][c]) p=fa[p];
            if (!p) p=1,tmp=0; else tmp=mx[p]+1,p=son[p][c];
        }
        ans+=sum[fa[p]]+1ll*(tmp-mx[fa[p]])*right[p];
    }
}

int main(){
    freopen("bzoj4566.in","r",stdin);
    freopen("bzoj4566.out","w",stdout);
    scanf("%s",s1+1); n=strlen(s1+1);
    scanf("%s",s2+1); m=strlen(s2+1);
    rep(i,1,n) ext(s1[i]-'a');
    radix(); solve(); printf("%lld
",ans);
    return 0;
}