BZOJ1195 [HNOI2006]最短母串 AC自动机 bfs

欢迎访问~原文出处——博客园-zhouzhendong

去博客园看该题解

---

传送门 - BZOJ1195

---

题意概括

　　给出一堆串，然后求一个包含这些串的所有串的最短的中的字典序最小的。

---

题解

　　先造一个AC自动机，多模匹配嘛。

　　然后bfs在AC自动机上面走，两维状态，dis[i][j]表示已经走到过的串状态为i，在AC自动机上面的位置为j的最短距离。

　　然后这题居然要卡空间！

　　坑死了。

　　然后用了short

　　wa掉了。

　　发现short实在小的可怜，然后把几个大的数组的有必要的一个开了int，然后30MB卡着限制过去了~

　　开心！！

---

代码

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cmath>
using namespace std;
struct Trie{
    int e,fail,next[];
    void init(){
        e=fail=;
        memset(next,,sizeof next);
    }
}tree[];
int cnt;
void AC_Prepare(){
    cnt=;
    tree[].init(),tree[].init();
    for (int i=;i<;i++)
        tree[].next[i]=;
}
void add(char ch[],int bh){
    int len=strlen(ch),rt=,t;
    for (int i=;i<len;i++){
        t=ch[i]-'A';
        if (!tree[rt].next[t]){
            tree[++cnt].init();
            tree[rt].next[t]=cnt;
        }
        rt=tree[rt].next[t];
    }
    tree[rt].e|=<<bh;
}
void build_AC(){
    int q[],head=,tail=,rt,son,k;
    tree[].fail=,q[++tail]=;
    while (head<tail){
        rt=q[++head];
        for (int i=;i<;i++){
            son=tree[rt].next[i];
            if (!son){
                tree[rt].next[i]=tree[tree[rt].fail].next[i];
                continue;
            }
            k=tree[rt].fail;
            while (!tree[k].next[i])
                k=tree[k].fail;
            tree[son].fail=tree[k].next[i];
            tree[son].e|=tree[tree[k].next[i]].e;
            q[++tail]=son;
        }
    }
}
int n,head,tail,pre[<<][];
short dis[<<][];
bool f[<<][];
char str[],chosen[<<][],ansstr[];
struct Queue{
    short x,y;
    void push(int a,int b){
        x=a,y=b;
    }
    void pushout(int &a,int &b){
        a=x,b=y;
    }
}q[];
int main(){
    scanf("%d",&n);
    AC_Prepare();
    for (int i=;i<n;i++){
        scanf("%s",str);
        add(str,i);
    }
    build_AC();
    memset(f,,sizeof f);
    memset(dis,,sizeof dis);
    memset(q,,sizeof q);
    head=tail=;
    q[++tail].push(,);
    dis[][]=,f[][]=,pre[][]=;
    int ans=-,x,y,x_,y_;
    while (head<tail){
        q[++head].pushout(x,y);
        for (int i=;i<;i++){
            y_=tree[y].next[i];
            x_=x|tree[y_].e;
            if (f[x_][y_])
                continue;
            dis[x_][y_]=dis[x][y]+;
            chosen[x_][y_]=i;
            pre[x_][y_]=head;
            f[x_][y_]=;
            q[++tail].push(x_,y_);
            if (x_==(<<n)-){
                ans=tail;
                break;
            }
        }
        if (ans!=-)
            break;
    }
    q[ans].pushout(x,y);
    int ansdis=dis[x][y],xnow,ynow;
    for (int j=ansdis,i=ans;j>=&&i;j--,i=pre[xnow][ynow]){
        q[i].pushout(xnow,ynow);
        ansstr[j]=chosen[xnow][ynow]+'A';
    }
    for (int i=;i<=ansdis;i++)
        printf("%c",ansstr[i]);
    return ;
}