洛谷 P2322 最短母串问题 状压+AC自动机

洛谷 P2322 最短母串问题 状压+AC自动机

题目大意：给你n个串，让你求包含这n个串的最短母串

首先，看到多串，就大概知道要用AC自动机了。然后再想，我们怎么将每个串加入到我们所求的母串上面去。虽然n<12,但是我们无法对每一种情况进行dfs，所以这种想法需要抛弃。

在trie树上，我们肯定需要遍历每一个节点，那么我们可以对每一个单词的结尾进行一个标记。怎么标记呢？（注意了，状压开始）我们对第i个单词的结尾建立一个state数组，state [now] = 1<<(i-1) ,即当前节点的编号now下，记录一个状态 2^(i-1),至于为什么是这个值，我们之后会有妙用。

之后我们在进行AC自动机的fail指针的建立的时候，顺便也把state的值传下去。

接着，就进行答案的bfs啦

我们需要开两个队列，一个记录节点，一个记录当前的状态，在开一个vis二维数组，vis[i] [j] 代表到第i个节点时处于 j 状态时候，是否之前来过。用一个st变量代表当前的状态

在bfs的时候，我们每到一个节点，就将st状态 or 该节点的state值（st | state[now]），记为当前到达此节点的状态，然后我们用vis数组存下来，记录为已到过，直到最后 st==1<<n-1，就说明我们这个状态已经包含了所有的单词了，因为是bfs，所以此时即为答案;

注意这里的状压转换是利用位运算的，0|1=1，1|0=1，1|1=0，0|0=0

所以每次 or 就代表有一个单词加入了

上代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=605;
int n,state[maxn],vis[maxn][1<<12|1],tot=0,cnt1=0,cnt2=0;
int trie[maxn][26],fail[maxn],ans[6500000],fa[6500000];
char str[13][51],anstr[maxn];
void build(char stt[],int p){
    int len=strlen(stt),root=0;
    for(int i=0;i<len;i++){
        int id=stt[i]-'A';
        if(!trie[root][id])
            trie[root][id]=++tot;
        root=trie[root][id];
    }
    state[root]|=1<<(p-1);//为什么这里是|，是为了保证多个相同的单词存在同一个状态里
}
void build_AC(){
    queue<int> Q;
    for(int i=0;i<26;i++)
        if(trie[0][i])
            Q.push(trie[0][i]),fail[trie[0][i]]=0;
    while (!Q.empty()){
        int x=Q.front();
        Q.pop();
        for(int i=0;i<26;i++)
        if(trie[x][i]){
            Q.push(trie[x][i]),fail[trie[x][i]]=trie[fail[x]][i];
            state[trie[x][i]]|=state[trie[fail[x]][i]];
        }
        else 
            trie[x][i]=trie[fail[x]][i];
    }
}
void getans(){
    queue<int> Q1,Q2;
    Q1.push(0),Q2.push(0);
    vis[0][0]=1;
    int id=0;
    while (!Q1.empty())
    {
        int now=Q1.front(),st=Q2.front();
        Q1.pop(),Q2.pop();
        if(st==((1<<n)-1)){
            while (id){
                anstr[++cnt2]=(char)(ans[id]+'A');
                id=fa[id];
            }
            return;
        }
        for(int i=0;i<26;i++)
        if(!vis[trie[now][i]][st|state[trie[now][i]]]){
            vis[trie[now][i]][st|state[trie[now][i]]]=1;
            Q1.push(trie[now][i]);
            Q2.push(st|state[trie[now][i]]);
            fa[++cnt1]=id;
            ans[cnt1]=i;
        }
        ++id;
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",str[i]);
        build(str[i],i);
    }
    build_AC();
    getans();
    for(int i=cnt2;i>0;i--)
        printf("%c",anstr[i]);
    return 0;
}