hdu3247 深刻理解ac自动机+bfs预处理最短路+状压DP

这题卡了我好久啊,蛋疼死我了,一开始还搜了个有错还能ac的代码。。。最后看着kuangbin大神的代码慢慢单步执行理解我写的哪里不对,然后发现他的代码也有点小bug= =,最后改了改他的就默默收藏起来了

贴出来是想着谁看到了的话就放心拿去对拍或者学习吧,我叫雷锋(虽然没人会看到的样子

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<iomanip>
#include<vector>
#include<set>
#include<map>
#include<queue>

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;

#define rep(i,k,n) for(int i=(k);i<=(n);i++)
#define rep0(i,n) for(int i=0;i<(n);i++)
#define red(i,k,n) for(int i=(k);i>=(n);i--)
#define sqr(x) ((x)*(x))
#define clr(x,y) memset((x),(y),sizeof(x))
#define pb push_back
#define mod 1000000007
const int INF=0x3f3f3f3f;
int pos[11];
struct Trie
{
    int next[60010][2],fail[60010],end[60010];
    int root,L;
    int newnode()
    {
        for(int i=0;i<2;i++)
            next[L][i]=-1;
        end[L]=0;
        return L++;
    }
    void init()
    {
        L=0;
        root=newnode();
    }
    int insert(char buf[],int id)
    {
        int len=strlen(buf);
        int now=root;
        for(int i=0;i<len;i++)
        {
            if(next[now][buf[i]-'0']==-1)
                next[now][buf[i]-'0']=newnode();
            now=next[now][buf[i]-'0'];
        }
        end[now]=id;
        return now;
    }
    void build()
    {
        queue<int> Q;
        fail[root]=root;
        for(int i=0;i<2;i++)
        {
            if(next[root][i]==-1)
                next[root][i]=root;
            else
            {
                fail[next[root][i]]=root;
                Q.push(next[root][i]);
            }
        }
        while(!Q.empty())
        {
            int now=Q.front();
            Q.pop();
            end[now]|=end[fail[now]];
            for(int i=0;i<2;i++)
                if(next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];
                else
                {
                    fail[next[now][i]]=next[fail[now]][i];
                    Q.push(next[now][i]);
                }
        }
    }
    int g[11][11];
    int dp[1024][11];
    int cnt;
    int dis[60010];

    void bfs(int k)
    {
        queue<int>q;
        memset(dis,-1,sizeof(dis));
        dis[pos[k]]=0;
        q.push(pos[k]);
        while(!q.empty())
        {
            int now=q.front();
            q.pop();
            for(int i=0;i<2;i++)
            {
                int tmp=next[now][i];
                if(dis[tmp]<0 && end[tmp]>=0)
                {
                    dis[tmp]=dis[now]+1;
                    q.push(tmp);
                }
            }
        }
        for(int i=0;i<cnt;i++)
            g[k][i]=dis[pos[i]];
    }

    int solve(int n)
    {
        pos[0]=0;
        cnt=n+1;
        for(int i=0;i<cnt;i++)bfs(i);

        for(int i=0;i<(1<<n);i++)
            for(int j=0;j<cnt;j++)
            dp[i][j]=INF;
        dp[0][0]=0;
        for(int i=0;i<(1<<n);i++)
            for(int j=0;j<cnt;j++)
            if(dp[i][j]<INF)
        {
            for(int k=0;k<cnt;k++)
            {
                if(g[j][k]<0 || j==k)continue;
                dp[i|end[pos[k]]][k]=min(dp[i|end[pos[k]]][k],dp[i][j]+g[j][k]);
            }
        }
        int ans=INF;
        for(int j=0;j<cnt;j++)
            ans=min(ans,dp[(1<<n)-1][j]);
        return ans;
    }
};
Trie ac;
char buf[1010];

int main()
{
	int n,m;
	while(scanf("%d%d",&n,&m),n||m)
    {
        ac.init();
        for(int i=0;i<n;i++)
        {
            scanf("%s",buf);
            pos[i+1]=ac.insert(buf,1<<i);
        }
        for(int i=0;i<m;i++)
        {
            scanf("%s",buf);
            ac.insert(buf,-1);
        }
        ac.build();
        printf("%d\n",ac.solve(n));
    }

	return 0;
}


你可能感兴趣的:(hdu3247 深刻理解ac自动机+bfs预处理最短路+状压DP)