[BZOJ3238][Ahoi2013]差异(后缀自动机)

题目:

我是超链接

题解:

后缀自动机的Parent指针反向以后可以形成一个树结构,和AC自动机的Fail树是类似的。这棵树有一个性质:两个子串的最长公共后缀所代表的状态,是这两个子串代表的状态在Parent树上的LCA位置。

原因也很简单,首先顺着Parent指针往上跳达到的就一定是一个串的后缀,并且跳的步数越少后缀越长。那么用最少的步数让两个状态到达一个节点,这个节点肯定就是最长的公共后缀了。

对于这道题目来说,要求的是最长公共前缀,我们可以把字符串倒过来,建立一棵parent树,这样树的最长公共后缀就是原串的最长公共前缀,然后对每个lca加贡献就好了

代码:

#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
const int N=2000005;
int ch[N][30],fa[N],step[N],p,np,q,nq,last,cnt;
LL ans,size[N],dis[N],val[N];
char st[N];int point[N],nxt[N],v[N],tot,c[N];
void addline(int x,int y,int z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void insert(int c)
{
    p=last; last=np=++cnt;
    step[np]=step[p]+1; size[np]=1;
    while (p && !ch[p][c]) ch[p][c]=np,p=fa[p];
    if (!p) {fa[np]=1;return;}
    q=ch[p][c];
    if (step[q]==step[p]+1){fa[np]=q;return;}
    nq=++cnt; step[nq]=step[p]+1;
    memcpy(ch[nq],ch[q],sizeof(ch[q]));
    fa[nq]=fa[q]; fa[q]=fa[np]=nq;
    while (ch[p][c]==q) ch[p][c]=nq,p=fa[p];
}
void dfs(int x,int fa)
{
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        dis[v[i]]=dis[x]+c[i];
        dfs(v[i],x);
        val[x]+=(LL)size[x]*size[v[i]]; size[x]+=size[v[i]];
      }
}
int main()
{
    scanf("%s",st+1);
    int l=strlen(st+1);
    last=cnt=1;
    for (int i=l;i>=1;i--) insert(st[i]-'a'); 
    for (int i=2;i<=cnt;i++)
    {
        int vv=step[i]-step[fa[i]];
        addline(fa[i],i,vv);
    }
    dfs(1,0);
    for (int i=1;i<=cnt;i++) ans+=(LL)dis[i]*val[i];
    LL sum=(LL)l*(l+1);
    sum=sum/2*(l-1);ans*=2;
    printf("%lld",sum-ans);
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值