洛谷p2322 ac自动机+最短路+dfs

ac自动机建立fail指针，求出每个点能覆盖的全部串种类，bfs求出最短母串长度（最短路），dfs求出路径输出。

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <queue>
#include <cstring>
#include <algorithm>
#include <string>
using namespace std;

const int maxn=700;
struct node{
    int x,y;
};
struct Tire{
    int nxt[maxn][26],fail[maxn];
    int dp[maxn][5000];//dp[i][j],在第i个状态点，走过了j代表的种类数，一共用了多少步
    int cnt,rt;
    int S[maxn],ans[maxn][5000];
    int newnode()
    {
        for(int i=0;i<26;i++)
        {
            nxt[cnt][i]=-1;
        }
        S[cnt++]=0;
        return cnt-1;
    }
    void init()
    {
        cnt=0;
        rt=newnode();
    }
    void ins(string s,int id)
    {
        int len=s.size();
        int now=rt;
        for(int i=0;i<len;i++)
        {
            if(nxt[now][s[i]-'A']==-1)
            {
                nxt[now][s[i]-'A']=newnode();
            }
            now=nxt[now][s[i]-'A'];
        }
        S[now]|=(1<<id);
    }
    void build()
    {
        queue<int>q;
        fail[rt]=rt;
        for(int i=0;i<26;i++)
        {
            if(nxt[rt][i]==-1)
            {
                nxt[rt][i]=rt;
            }
            else
            {
                fail[nxt[rt][i]]=rt;
                q.push(nxt[rt][i]);
            }
        }
        while(!q.empty())
        {
            int now=q.front();
            q.pop();
            for(int i=0;i<26;i++)
            {
                if(nxt[now][i]==-1)
                {
                    nxt[now][i]=nxt[fail[now]][i];
                }
                else
                {
                    fail[nxt[now][i]]=nxt[fail[now]][i];
                    q.push(nxt[now][i]);
                }
            }
        }
    }
    int dfs(int x,int y,int len,int n)
    {
        if(y==(1<<n)-1)return dp[x][y];
        for(int i=0;i<26;i++)
        {
            int xx=nxt[x][i];
            if(xx!=-1&&dp[x][y]+1==dp[xx][y|S[xx]])
            {
                if(dfs(xx,y|S[xx],len,n)==len)
                {
                    ans[x][y]=i;
                    return len;
                }
            }
        }
    }
    void query(int n)
    {
        for(int i=0;i<cnt;i++)
        {
            int tt=fail[i];//统计取这个路径能够取到的单词的种类
            while(tt)
            {
                S[i]|=S[tt];
                tt=fail[tt];
            }
            for(int j=0;j<(1<<n);j++)
            {
                dp[i][j]=1000000;//初始化dp为最大值，跑最短路时进行更新
            }
        }
        dp[0][0]=0;//从0,0点开始走，进行最短路（bfs）要求出覆盖所有字符串的最小字符串长度
        queue<node>q;
        q.push(node{0,0});
        while(!q.empty())
        {
            node tt=q.front();
            q.pop();
            for(int i=0;i<26;i++)
            {
                if(nxt[tt.x][i])
                if(dp[tt.x][tt.y]+1<dp[nxt[tt.x][i]][tt.y|S[nxt[tt.x][i]]])//如果从当前状态点+1，小于到达下一个状态点的步数的话，我们就将其压入队中
                {
                    dp[nxt[tt.x][i]][tt.y|S[nxt[tt.x][i]]]=dp[tt.x][tt.y]+1;
                    q.push(node{nxt[tt.x][i],tt.y|S[nxt[tt.x][i]]});
                }
            }
        }
        int len=1000000;
        for(int i=0;i<cnt;i++)
            len=min(len,dp[i][(1<<n)-1]);//求出最小长度，（1<<n）-1这个位置说明全部串都经过
        dfs(0,0,len,n);//dfs求路径
        int x=0,y=0;
        for(int i=1;i<=len;i++)
        {
            printf("%c",ans[x][y]+'A');
            int cjr=nxt[x][ans[x][y]];
            x=cjr;
            y|=S[cjr];
        }
    }
};
Tire ac;
string s;
int main()
{
        int n;
        scanf("%d",&n);
        ac.init();
        for(int i=0;i<n;i++)
        {
             cin>>s;
             ac.ins(s,i);
        }
        ac.build();
        ac.query(n);
    return 0;
}