【BZOJ3238】差异（AHOI2013）-后缀自动机+树形DP

最新推荐文章于 2019-04-15 20:00:05 发布

原创最新推荐文章于 2019-04-15 20:00:05 发布 · 256 阅读

1 ·

CC 4.0 BY-SA版权

动态规划-树形DP 同时被 2 个专栏收录

35 篇文章

订阅专栏

字符串-后缀自动机（SAM）

9 篇文章

订阅专栏

本文介绍了一种解决字符串匹配问题的方法——利用后缀自动机结合树形动态规划（树形DP）。通过翻转字符串将问题转化为求最长公共后缀，再借助后缀自动机快速找到最长公共子串。文章提供了完整的代码实现，并详细解释了如何构建后缀自动机和进行DP的状态转移。

测试地址：差异
做法：本题需要用到后缀自动机+树形DP。
我们把字符串翻转，那么原串两个后缀的最长公共前缀就变成了两个前缀的最长公共后缀。注意到在后缀自动机上，两个子串的最长公共后缀就是它们在后缀链接上的LCA，那么我们先建出后缀自动机，然后在后缀链接上DP，对于每个点，求以这个点为LCA的所有前缀对对答案的贡献。状态转移方程就不写了，详见代码，算法的总时间复杂度为 $O(n)$ ，可以通过此题。
以下是本人代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,tot=1,last,pre[1000010]={0},ch[1000010][26]={0};
int first[1000010]={0},tote=0;
ll len[1000010],f[1000010]={0},totlen[1000010]={0},ans=0;
char s[500010];
struct edge
{
    int v,next;
}e[1000010];

void init()
{
    scanf("%s",s);
    n=strlen(s);
    for(int i=0;i<n-i-1;i++)
        swap(s[i],s[n-i-1]);
}

void extend(char c)
{
    int p,q,np,nq;
    np=++tot;
    len[np]=len[last]+1;
    f[np]=1,totlen[np]=len[np];
    p=last;
    while(p&&!ch[p][c-'a']) ch[p][c-'a']=np,p=pre[p];
    if (!p) pre[np]=1;
    else
    {
        q=ch[p][c-'a'];
        if (len[p]+1==len[q]) pre[np]=q;
        else
        {
            nq=++tot;
            len[nq]=len[p]+1;
            pre[nq]=pre[q];
            for(int i=0;i<26;i++)
                ch[nq][i]=ch[q][i];
            while(p&&ch[p][c-'a']==q) ch[p][c-'a']=nq,p=pre[p];
            pre[q]=pre[np]=nq;
        }
    }
    last=np;
}

void insert(int a,int b)
{
    e[++tote].v=b;
    e[tote].next=first[a];
    first[a]=tote;
}

void build()
{
    last=tot=1;
    for(int i=0;i<n;i++)
        extend(s[i]);
    for(int i=2;i<=tot;i++)
        insert(pre[i],i);
}

void dp(int v)
{
    ll sum=0,sumlen=0;
    for(int i=first[v];i;i=e[i].next)
    {
        dp(e[i].v);
        ans-=(sum*f[e[i].v]*len[v])<<1;
        ans+=f[e[i].v]*sumlen+sum*totlen[e[i].v];
        sum+=f[e[i].v];
        sumlen+=totlen[e[i].v];
    }
    if (f[v])
    {
        ans-=(sum*len[v])<<1;
        ans+=sumlen+sum*totlen[v];
    }
    f[v]+=sum,totlen[v]+=sumlen;
}

int main()
{
    init();
    build();
    dp(1);
    printf("%lld",ans);

    return 0;
}