AC日记——[Wc2008]游览计划 bzoj 2595

2595

思路：

　　状压DP+spfa转移+dfs输出路径；

　　或者，斯坦纳树算法模板；

来，上代码：

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define INF 0x7ffffff

struct NodeType {
    int x,y,s;
};
struct NodeType pre[][][];

const int dx[]={,-,,,};
const int dy[]={,,,,-};

int n,m,map[][],num,bi[],all,dp[][][],ans;

bool if_[][],cho[][];

queue<NodeType>que;

inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}

void read()
{
    in(n),in(m),bi[]=;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            in(map[i][j]);
            if(!map[i][j]) num++;
        }
    }
    for(int i=;i<=num;i++) bi[i]=bi[i-]<<;
    all=bi[num]-;
}

void spfa(int f)
{
    while(!que.empty())
    {
        NodeType now=que.front();que.pop(),if_[now.x][now.y]=false;
        for(int i=;i<=;i++)
        {
            int x=now.x+dx[i],y=now.y+dy[i];
            if(x>&&x<=n&&y>&&y<=m)
            {
                if(dp[x][y][f]>dp[now.x][now.y][f]+map[x][y])
                {
                    dp[x][y][f]=dp[now.x][now.y][f]+map[x][y];
                    pre[x][y][f]=(NodeType){now.x,now.y,now.s};
                    if(!if_[x][y]) que.push((NodeType){x,y,f}),if_[x][y]=true;
                }
            }
        }
    }
}

void dfs(int x,int y,int s)
{
    if(x>INF||pre[x][y][s].s==) return ;
    cho[x][y]=true;
    dfs(pre[x][y][s].x,pre[x][y][s].y,pre[x][y][s].s);
    if(x==pre[x][y][s].x&&y==pre[x][y][s].y) dfs(x,y,s-pre[x][y][s].s);
}

void solve()
{
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            for(int v=;v<=all;v++) dp[i][j][v]=INF;
        }
    }
    int F=;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            if(!map[i][j]) dp[i][j][bi[F++]]=;
        }
    }
    for(int S=;S<=all;S++)
    {
        for(int i=;i<=n;i++)
        {
            for(int j=;j<=m;j++)
            {
                for(int s=S&(S-);s;s=S&(s-))
                {
                    int tmp=dp[i][j][s]+dp[i][j][S-s]-map[i][j];
                    if(tmp<dp[i][j][S])
                    {
                        dp[i][j][S]=tmp;
                        pre[i][j][S]=(NodeType){i,j,s};
                    }
                }
                if(dp[i][j][S]<INF) que.push((NodeType){i,j,S}),if_[i][j]=true;
            }
        }
        spfa(S);
    }
    int ex,ey;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            if(!map[i][j])
            {
                ex=i,ey=j,ans=dp[i][j][all];
                break;
            }
        }
    }
    dfs(ex,ey,all);
}

void print()
{
    printf("%d\n",ans);
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            if(!map[i][j]) putchar('x');
            else if(cho[i][j]) putchar('o');
            else putchar('_');
        }
        putchar('\n');
    }
}

int main()
{
    read();
    solve();
    print();
    return ;
}