1. 程式人生 > >[HAOI2016]找相同字元(SAM+DP)

[HAOI2016]找相同字元(SAM+DP)

感覺很水。
因為SAM上一個點的子樹大小代表這個點所表示字首的出現次數。
建出廣義字尾自動機之後。在\(parent\)樹上跑\(DP\),維護\(size[i][1]\),和\(size[i][0]\)代表i的子樹中有多少第一個串的結束節點和第二個串的結束節點,然後答案就是\(size[i][0]*size[i][1]*(len[i]-len[fa[i]])\)

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=801000;
char s1[N],s2[N];
long long ans;
int cnt,head[N];
struct edge{
    int to,nxt;
}e[N];
void add(int u,int v){
    cnt++;
    e[cnt].nxt=head[u];
    e[cnt].to=v;
    head[u]=cnt;
}
struct SAM{
    int tot,u,fa[N],size[N][3],len[N],trans[N][27];
    void init(){tot=u=1;}
    void rebuild(){u=1;}
    void ins(int c,int k){
        int x=++tot;len[x]=len[u]+1;size[x][k]=1;
        for(;u&&trans[u][c]==0;u=fa[u])trans[u][c]=x;
        if(u==0)fa[x]=1;
        else{
            int v=trans[u][c];
            if(len[u]+1==len[v])fa[x]=v;
            else{
                int w=++tot;
                len[w]=len[u]+1;
                fa[w]=fa[v];
                memcpy(trans[w],trans[v],sizeof(trans[w]));
                fa[x]=fa[v]=w;
                for(;u&&trans[u][c]==v;u=fa[u])trans[u][c]=w;
            }
        }
        u=x;
    }
    void work(int u){
        //cout<<u<<endl;
        for(int i=head[u];i;i=e[i].nxt){
            int v=e[i].to;
            work(v);
            size[u][1]+=size[v][1];
            size[u][2]+=size[v][2];
        }
        ans+=(long long)size[u][1]*size[u][2]*(len[u]-len[fa[u]]);
    }
}sam;
int main(){
    scanf("%s",s1+1);
    scanf("%s",s2+1);
    sam.init();
    int len1=strlen(s1+1);
    for(int i=1;i<=len1;i++)sam.ins(s1[i]-'a'+1,1);
    sam.rebuild();
    int len2=strlen(s2+1);
    for(int i=1;i<=len2;i++)sam.ins(s2[i]-'a'+1,2);
    for(int i=1;i<=sam.tot;i++)add(sam.fa[i],i);
    sam.work(1);
    printf("%lld",ans);
    return 0;
}