[BZOJ4566][Haoi2016]找相同字符（后缀自动机）

最新推荐文章于 2019-10-16 09:07:28 发布

Clove_unique

最新推荐文章于 2019-10-16 09:07:28 发布

阅读量859

点赞数

分类专栏：题解后缀自动机省选

本文链接：https://blog.csdn.net/Clove_unique/article/details/68482178

版权

题解同时被 3 个专栏收录

1097 篇文章 4 订阅

订阅专栏

省选

211 篇文章 0 订阅

订阅专栏

后缀自动机

18 篇文章 0 订阅

订阅专栏

本文介绍了如何利用后缀自动机解决找相同字符的问题，通过将两个字符串连接并插入分隔符，然后在自动机上统计每个状态对应的字符出现次数，结合状态转移步数计算答案。题解指出直接使用SA（suffix array）可能不是最佳方法。

摘要由CSDN通过智能技术生成

题目描述

传送门

题解

将两个串接在一起建立后缀自动机，中间加一个分隔符
对于自动机上的每一个点，统计其right集合中在第一个串中的数量cnt1，在第二个串中的数量cnt2
对于某一个点合法的长度是[step(pre(i))+1,step(i)]
答案累加就是cnt1*cnt2*(step(i)-step(pre(i)))
感觉这题sa是不大好做的

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 800005
#define LL long long

char s[N],s2[N];
int n,n2,dvd,root,last,sz,p,np,q,nq;
int ch[N][30],pre[N],step[N],c[N],pt[N],cnt1[N],cnt2[N];
LL ans;

void extend()
{
    root=last=++sz;
    for (int i=0;i<n;++i)
    {
        int x=s[i]-'a';
        p=last;np=++sz;last=np;
        step[np]=step[p]+1;
        while (p&&!ch[p][x])
        {
            ch[p][x]=np;
            p=pre[p];
        }
        if (!p) pre[np]=root;
        else
        {
            q=ch[p][x];
            if (step[q]==step[p]+1) pre[np]=q;
            else
            {
                nq=++sz;
                step[nq]=step[p]+1;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                pre[nq]=pre[q];
                while (ch[p][x]==q)
                {
                    ch[p][x]=nq;
                    p=pre[p];
                }
                pre[q]=pre[np]=nq;
            }
        }
    }
}
int main()
{
    scanf("%s",s);dvd=n=strlen(s);
    s[n++]='$';
    scanf("%s",s2);n2=strlen(s2);
    for (int i=0;i<n2;++i) s[n++]=s2[i];
    extend();
    for (int i=1;i<=sz;++i) ++c[step[i]];
    for (int i=1;i<=sz;++i) c[i]+=c[i-1];
    for (int i=sz;i>=1;--i) pt[c[step[i]]--]=i;
    p=root;
    for (int i=0;i<n;++i)
    {
        int x=s[i]-'a';
        p=ch[p][x];
        if (step[p]<=dvd) ++cnt1[p];
        else ++cnt2[p];
    }
    for (int i=sz;i>=1;--i)
    {
        p=pt[i];
        cnt1[pre[p]]+=cnt1[p];
        cnt2[pre[p]]+=cnt2[p];
    }
    for (int i=1;i<=sz;++i)
    {
        if (s[step[i]-1]=='$') continue;
        ans+=(LL)cnt1[i]*cnt2[i]*(step[i]-step[pre[i]]);
    }
    printf("%lld\n",ans);
}