【BZOJ 2595】2595: [Wc2008]游览计划 （状压DP+spfa，斯坦纳树？）

  1 #include<cstdio>
  2 #include<cstdlib>
  3 #include<cstring>
  4 #include<iostream>
  5 #include<algorithm>
  6 #include<queue>
  7 using namespace std;
  8 #define INF 0xfffffff
  9 
 10 int a[15][15],num[15][15],f[15][15][1500];
 11 
 12 struct node {int x,y,d;};
 13 node g[15][15][1500];
 14 queue<node > q;
 15 bool inq[15][15];
 16 
 17 int bx[6]={0,1,0,-1,0},
 18     by[6]={0,0,1,0,-1};
 19 
 20 void dfs(int x,int y,int k)
 21 {
 22     if(!k) return;inq[x][y]=1;
 23     dfs(g[x][y][k].x,g[x][y][k].y,g[x][y][k].d);
 24     if(g[x][y][k].x==x&&g[x][y][k].y==y) dfs(x,y,k-g[x][y][k].d);
 25 }
 26 
 27 int main()
 28 {
 29     int n,m,cnt=0;
 30     scanf("%d%d",&n,&m);
 31     memset(f,63,sizeof(f));
 32     for(int i=1;i<=n;i++)
 33      for(int j=1;j<=m;j++)
 34      {
 35          scanf("%d",&a[i][j]);
 36          if(a[i][j]==0)
 37          {
 38              num[i][j]=++cnt;
 39              f[i][j][1<<cnt-1]=0;
 40              g[i][j][1<<cnt-1].x=g[i][j][1<<cnt-1].y=g[i][j][1<<cnt-1].d=0;
 41          }
 42      }
 43     for(int i=1;i<=n;i++)
 44      for(int j=1;j<=m;j++) f[i][j][0]=0;
 45     for(int k=1;k<=(1<<cnt)-1;k++)
 46     {
 47         memset(inq,0,sizeof(inq));
 48         for(int ss=k;ss;ss=(ss-1)&k)
 49         {
 50             for(int i=1;i<=n;i++)
 51              for(int j=1;j<=m;j++) //if(a[i][j]==0)
 52              {
 53                  if(f[i][j][k]>f[i][j][ss]+f[i][j][k-ss]-a[i][j])
 54                 {
 55                     f[i][j][k]=f[i][j][ss]+f[i][j][k-ss]-a[i][j];
 56                     node nw;
 57                     nw.x=i;nw.y=j;nw.d=ss;
 58                     g[i][j][k]=nw;
 59                 }
 60                 if(f[i][j][k]<INF) {node nw;nw.x=i;nw.y=j;nw.d=f[i][j][k];inq[i][j]=1;q.push(nw);}
 61              }
 62             
 63         }
 64         while(!q.empty())
 65         {
 66             node x=q.front();
 67             for(int i=1;i<=4;i++)
 68             {
 69                 int nx=x.x+bx[i],ny=x.y+by[i];
 70                 if(nx<1||nx>n||ny<1||ny>m) continue;
 71                 if(f[nx][ny][k]>f[x.x][x.y][k]+a[nx][ny])
 72                 {
 73                     f[nx][ny][k]=f[x.x][x.y][k]+a[nx][ny];
 74                     node nw;
 75                     nw.x=nx;nw.y=ny;//nw.d=f[nx][ny][k];
 76                     // g[nx][ny][k]=g[x.x][x.y][k];
 77                     g[nx][ny][k].x=x.x;g[nx][ny][k].y=x.y;g[nx][ny][k].d=k;
 78                     if(!inq[nx][ny])
 79                     {
 80                         inq[nx][ny]=1;
 81                         q.push(nw);
 82                     }
 83                 }
 84             }
 85             q.pop();inq[x.x][x.y]=0;
 86         }
 87     }
 88     memset(inq,0,sizeof(inq));
 89     bool ok=0;
 90     for(int i=1;i<=n;i++)
 91     {
 92         for(int j=1;j<=m;j++) if(a[i][j]==0)
 93         {
 94             printf("%d\n",f[i][j][(1<<cnt)-1]);
 95             dfs(i,j,(1<<cnt)-1);
 96             ok=1;break;
 97         }
 98         if(ok) break;
 99     }
100      
101     for(int i=1;i<=n;i++)
102     {
103         for(int j=1;j<=m;j++)
104         {
105             if(a[i][j]==0) printf("x");
106             else if(inq[i][j]) printf("o");
107             else printf("_");
108         }
109         printf("\n");
110     }
111     return 0;
112 }