【BZOJ4032】[HEOI2015]最短不公共子串(后缀自动机,序列自动机)

【BZOJ4032】[HEOI2015]最短不公共子串(后缀自动机,序列自动机)

题面

BZOJ
洛谷

题解

数据范围很小,直接暴力构建后缀自动机和序列自动机,然后直接在两个自动机上进行\(bfs\),找到的第一个不同时存在的节点就直接输出就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define MAX 4040
struct SAM
{
    struct Node{int son[26],ff,len;}t[MAX];
    int tot,last;void init(){tot=last=1;}
    void extend(int c)
    {
        int p=last,np=++tot;last=tot;
        t[np].len=t[p].len+1;
        while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].ff;
        if(!p)t[np].ff=1;
        else
        {
            int q=t[p].son[c];
            if(t[q].len==t[p].len+1)t[np].ff=q;
            else
            {
                int nq=++tot;
                t[nq]=t[q];t[nq].len=t[p].len+1;
                t[q].ff=t[np].ff=nq;
                while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].ff;
            }
        }
    }
}A,B;
struct SQAM
{
    struct Node{int son[26];}t[MAX];
    int lst[26],pre[MAX],tot,last;
    void init(){tot=last=1;for(int i=0;i<26;++i)lst[i]=1;}
    void extend(int c)
    {
        int p=lst[c],np=++tot;
        pre[np]=p;
        for(int i=0;i<26;++i)
            for(int j=lst[i];j&&!t[j].son[c];j=pre[j])
                t[j].son[c]=np;
        lst[c]=np;
    }
}AA,BB;
char SA[MAX],SB[MAX];
int na,nb;
struct Node{int len,a,b;};
bool vis[MAX][MAX];
int bfs1(SAM &A,SAM &B)
{
    queue<Node> Q;Q.push((Node){0,1,1});
    memset(vis,0,sizeof(vis));vis[1][1]=true;
    while(!Q.empty())
    {
        Node p=Q.front();Q.pop();
        for(int i=0;i<26;++i)
        {
            int x=A.t[p.a].son[i];
            int y=B.t[p.b].son[i];
            if(x&&y)
            {
                if(vis[x][y])continue;
                vis[x][y]=true;Q.push((Node){p.len+1,x,y});
            }
            if(x&&!y)return p.len+1;
        }
    }
    return -1;  
}
int bfs2(SAM &A,SQAM &B)
{
    queue<Node> Q;Q.push((Node){0,1,1});
    memset(vis,0,sizeof(vis));vis[1][1]=true;
    while(!Q.empty())
    {
        Node p=Q.front();Q.pop();
        for(int i=0;i<26;++i)
        {
            int x=A.t[p.a].son[i];
            int y=B.t[p.b].son[i];
            if(x&&y)
            {
                if(vis[x][y])continue;
                vis[x][y]=true;Q.push((Node){p.len+1,x,y});
            }
            if(x&&!y)return p.len+1;
        }
    }
    return -1;  
}
int bfs3(SQAM &A,SAM &B)
{
    queue<Node> Q;Q.push((Node){0,1,1});
    memset(vis,0,sizeof(vis));vis[1][1]=true;
    while(!Q.empty())
    {
        Node p=Q.front();Q.pop();
        for(int i=0;i<26;++i)
        {
            int x=A.t[p.a].son[i];
            int y=B.t[p.b].son[i];
            if(x&&y)
            {
                if(vis[x][y])continue;
                vis[x][y]=true;Q.push((Node){p.len+1,x,y});
            }
            if(x&&!y)return p.len+1;
        }
    }
    return -1;  
}
int bfs4(SQAM &A,SQAM &B)
{
    queue<Node> Q;Q.push((Node){0,1,1});
    memset(vis,0,sizeof(vis));vis[1][1]=true;
    while(!Q.empty())
    {
        Node p=Q.front();Q.pop();
        for(int i=0;i<26;++i)
        {
            int x=A.t[p.a].son[i];
            int y=B.t[p.b].son[i];
            if(x&&y)
            {
                if(vis[x][y])continue;
                vis[x][y]=true;Q.push((Node){p.len+1,x,y});
            }
            if(x&&!y)return p.len+1;
        }
    }
    return -1;  
}
int main()
{
    scanf("%s%s",SA+1,SB+1);
    na=strlen(SA+1);nb=strlen(SB+1);
    A.init();B.init();AA.init();BB.init();
    for(int i=1;i<=na;++i)A.extend(SA[i]-97);
    for(int i=1;i<=nb;++i)B.extend(SB[i]-97);
    for(int i=1;i<=na;++i)AA.extend(SA[i]-97);
    for(int i=1;i<=nb;++i)BB.extend(SB[i]-97);
    printf("%d\n%d\n%d\n%d\n",bfs1(A,B),bfs2(A,BB),bfs3(AA,B),bfs4(AA,BB));
    return 0;
}
posted @ 2019-04-28 20:00 小蒟蒻yyb 阅读(...) 评论(...) 编辑 收藏