[BZOJ4566][Haoi2016]找相同字符 后缀自动机+dp

4566: [Haoi2016]找相同字符

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 1212  Solved: 694
[Submit][Status][Discuss]

Description

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

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

Input

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

Output

输出一个整数表示答案

Sample Input

aabb
bbaa

Sample Output

10

HINT

 

Source

时隔许久后重新获得了后缀自动机技能。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<cstdio>
#define maxn 200005
#define ll long long
using namespace std;
inline int read() {
    int x=0,f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';
    return x*f;
}
struct SUF {
    int sz[maxn*2],link[maxn*2],step[maxn*2],son[maxn*2][26],v[maxn*2],pos[maxn*2];
    int nxt[maxn*2],head[maxn*2],tto[maxn*2],tot;
    int cnt,last;
    SUF() {last=cnt=1;memset(head,-1,sizeof(head));}
    void extend(int c) {
        int p=last,np=last=++cnt;step[np]=step[p]+1;sz[np]=1;
        while(p&&!son[p][c]) son[p][c]=np,p=link[p];
        if(!p) link[np]=1;
        else {
            int q=son[p][c];
            if(step[q]==step[p]+1) link[np]=q;
            else {
                int nq=++cnt;
                memcpy(son[nq],son[q],sizeof(son[q]));
                link[nq]=link[q];link[q]=link[np]=nq;
                step[nq]=step[p]+1;
                while(son[p][c]==q&&p) son[p][c]=nq,p=link[p];
            }
        }
    }
    void pre() {
        for(int i=1;i<=cnt;i++) v[step[i]]++;
        for(int i=1;i<=cnt;i++) v[i]+=v[i-1];
        for(int i=cnt;i>=1;i--) pos[v[step[i]]--]=i;
        for(int i=cnt;i>=1;i--) sz[link[pos[i]]]+=sz[pos[i]];
    }
    ll f[maxn*2];
    void dfs(int x) {
        f[x]+=1ll*sz[x]*(step[x]-step[link[x]]);
        for(int i=head[x];i>=0;i=nxt[i]) {
            int to=tto[i];f[to]+=f[x];dfs(to);
        }
    }
    void add(int u,int v) {
        nxt[tot]=head[u];tto[tot]=v;head[u]=tot++;
    }
}t;
char ch[maxn];
int main() {
    scanf("%s",ch+1);int len=strlen(ch+1);
    for(int i=1;i<=len;i++) t.extend(ch[i]-'a');
    t.pre();
    scanf("%s",ch+1);len=strlen(ch+1);
    for(int i=1;i<=t.cnt;i++) if(t.link[i]) t.add(t.link[i],i);t.dfs(1);
    int now=1,l=0;ll ans=0;
    for(int i=1;i<=len;i++) {
        int to=ch[i]-'a';
        while(now&&!t.son[now][to]) now=t.link[now];
        if(!now) now=1,l=0;
        else l=min(l,t.step[now])+1,now=t.son[now][to];
        ans+=t.f[t.link[now]];ans+=1ll*(l-t.step[t.link[now]])*t.sz[now];
    }
    printf("%lld
",ans);
}