Bzoj4566 [Haoi2016]找相同字符

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 378  Solved: 209

Description

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

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

Input

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

Output

输出一个整数表示答案

Sample Input

aabb
bbaa

Sample Output

10

HINT

 

Source

字符串 广义后缀自动机

把两个字符串建到同一个SAM上，分别记录size。如果某个结点同时有两个串的size，那么可以更新答案

注意第23行，如果一个节点已经有了，就不新建结点了。虽然不知道为什么，但是这步似乎是必要的，不加就WA

/*by SilverN*/
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
#define LL long long
using namespace std;
const int mxn=800010;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct SAM{
    int t[mxn][26],fa[mxn],l[mxn];
    int sz[mxn][2],w[mxn],rk[mxn];
    int S,cnt;
    void init(){S=cnt=1;return;}
    int add(int c,int p,int bl){
        if(t[p][c] && l[t[p][c]]==l[p]+1)return t[p][c];
        int np=++cnt;
        l[np]=l[p]+1;
//      sz[np][bl]++;
        for(;p && !t[p][c];p=fa[p])t[p][c]=np;
        if(!p)fa[np]=S;
        else{
            int q=t[p][c];
            if(l[q]==l[p]+1){fa[np]=q;}
            else{
                int nq=++cnt;l[nq]=l[p]+1;
                memcpy(t[nq],t[q],sizeof t[q]);
                fa[nq]=fa[q];
                fa[q]=fa[np]=nq;
                for(; p&& t[p][c]==q;p=fa[p])t[p][c]=nq;
            }
        }
//      printf("np:%d %d
",np,fa[np]);
        return np;
    }
    void solve(int len){
        int i,j;
        for(i=1;i<=cnt;i++)w[l[i]]++;
        for(i=1;i<=len;i++)w[i]+=w[i-1];
        for(i=1;i<=cnt;i++)rk[w[l[i]]--]=i;
        LL ans=0;
        for(int i=cnt;i;i--){
            int t=rk[i];
//          printf("t:%d  %d
",t,fa[t]);
            sz[fa[t]][0]+=sz[t][0];
            sz[fa[t]][1]+=sz[t][1];
        }
        for(i=1;i<=cnt;i++){
            ans+=(LL)(l[i]-l[fa[i]])*sz[i][0]*sz[i][1];
        }
        printf("%lld
",ans);
        return;
    }
}sa;
char s1[mxn>>1],s2[mxn>>1];
int main(){
    int i,j;
    sa.init();
    scanf("%s%s",s1+1,s2+1);
    int n1=strlen(s1+1);
    int n2=strlen(s2+1);
    int id=sa.S;
    for(i=1;i<=n1;i++)id=sa.add(s1[i]-'a',id,0),sa.sz[id][0]++;
    id=sa.S;
    for(i=1;i<=n2;i++)id=sa.add(s2[i]-'a',id,1),sa.sz[id][1]++;
    sa.solve(max(n1,n2));
    return 0;
}