bzoj2780 广义后缀自动机+parent树+Dfs序+树状数组

最新推荐文章于 2019-06-19 18:39:00 发布

rainbow979

最新推荐文章于 2019-06-19 18:39:00 发布

阅读量1.2k

点赞数

分类专栏： BZOJ 后缀自动机文章标签： bzoj 广义后缀自动机 Dfs序

本文链接：https://blog.csdn.net/rainbow979/article/details/50448867

版权

BZOJ 同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

后缀自动机

2 篇文章 0 订阅

订阅专栏

本文介绍了一种用于处理多个字符串的广义后缀自动机算法。通过该算法可以高效地解决字符串匹配问题，特别是在需要查询一个字符串出现在哪些源字符串中的场景下。文章详细解释了如何构建广义后缀自动机，并提供了完整的代码实现。

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

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=2780
大意：给n个串，再给m个询问，询问一个字符串是出现在哪几个母串中。
bzoj2754的强化版。

广义后缀自动机：对于多个串建立后缀自动机，每次只要将last改为1就好了。

这题思路大体同bzoj2434。
但还是具体讲一讲。。。
首先建广义后缀自动机，对于每个节点，记录它属于哪几个串，挂到节点上。然后一样的建parent树，一样的Dfs序。
然后在广义后缀自动机上走询问，如果找不到答案就为0，否则将询问挂到当前节点上。
然后遍历广义后缀自动机，更新答案。

#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
struct sef { int ne,en; }h[400050];
int tt,first[400050];
int ans[400050],n,m;
int last,len;
int vis[400050];
char s[400050];
struct Tree {
    sef h[400050];
    int tot,first[400040];
    int To[400050],Out[400050],tim,d[400050];
    inline void setl(int x,int y) {
        h[++tot].ne=first[x]; h[tot].en=y;
        first[x]=tot;
    }
    inline void Dfs(int x) {
        int y;
        To[x]=++tim;    d[tim]=x;
        for (int i=first[x];y=h[i].en,i;i=h[i].ne) Dfs(y);
        Out[x]=tim;
    }
}G;
struct Query {
    sef h[400050];
    int tot,first[400050],w[400050];

    inline void setl(int x,int y,int v) {
        h[++tot].ne=first[x]; h[tot].en=y; w[tot]=v;
        first[x]=tot;
    }
}Q;

struct Fenwick {
    int d[400050];
    inline void Add(const int &x,const int &v) {
        for (int i=x;i<=G.tim;i+=i & (-i)) d[i]+=v;
    }
    inline int Ask(int x) {
        int r=0;
        for (int i=G.Out[x];i;i-=i & (-i)) r+=d[i];
        for (int i=G.To[x]-1;i;i-=i & (-i)) r-=d[i];
        return r;
    }
}fw;

struct SAM {
    map<int , int > son;
    int len,fa;
}C[400040];
int tot;

inline void setl(int x,int y)
{
    h[++tt].ne=first[x]; h[tt].en=y;
    first[x]=tt;
}
inline void Add(int x,int k)
{
    int p=last,np;
    if (!C[p].son[x]) {
        last=np=++tot;  C[np].len=C[p].len+1;
        for (;p&&!C[p].son[x];p=C[p].fa) C[p].son[x]=np;
        if (!p) C[np].fa=1;
        else {
            int q=C[p].son[x];
            if (C[p].len+1==C[q].len) C[np].fa=q;
            else {
                int nq=++tot;   C[nq]=C[q];
                C[nq].len=C[p].len+1;
                C[q].fa=C[np].fa=nq;
                for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
            }
        }
    } else {
        int q=C[p].son[x];
        if (C[p].len+1==C[q].len) last=q;
        else {
            int nq=++tot;   C[nq]=C[q];
            C[nq].len=C[p].len+1;
            C[q].fa=nq;
            for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
            last=nq;
        }
    }
    setl(last,k);
}
int main()
{
    scanf("%d%d",&n,&m);
    tot=1;
    for (int i=1;i<=n;++i) {
        scanf("%s",s);
        len=strlen(s);  last=1;
        for (int j=0;j<len;++j)
            Add(s[j],i);
    }
    for (int i=2;i<=tot;++i) G.setl(C[i].fa,i);
    G.Dfs(1);
    int x; bool boo;
    for (int i=1;i<=m;++i) {
        scanf("%s",s);
        len=strlen(s);
        x=1;    boo=1;
        for (int j=0;j<len;++j) {
            if (!C[x].son[s[j]]) { boo=0; break; }
            else x=C[x].son[s[j]];
        }
        if (boo) Q.setl(G.Out[x],i,x);
    }
    int y;
    for (int j=1;j<=tot;++j) {
        x=G.d[j];
        for (int i=first[x];y=h[i].en,i;i=h[i].ne) {
            fw.Add(j,1);
            if (vis[y]) fw.Add(vis[y],-1);
            vis[y]=j;
        }
        for (int i=Q.first[j];y=Q.h[i].en,i;i=Q.h[i].ne)
            ans[y]=fw.Ask(Q.w[i]);
    }
    for (int i=1;i<=m;++i) printf("%d\n",ans[i]);
}