WC2008 浏览计划

斯坦纳树

#include<cstdio>
#include<cstring>
#include<queue>
#define N 11
#define M 1100
#define INF 100000000
using namespace std;
int n,m,k,K;
int st[N][N];
int dp[N][N][M];
int pre[N][N][M];
bool vis[N][N][M],ok[N][N];
int map[N][N];
queue<int>q;
int h[]={1,-1,0,0};
int g[]={0,0,-1,1};
void init(){
    int i,j,p;
    k=0;
    memset(st,0,sizeof(st));
    memset(vis,0,sizeof(vis));
    memset(pre,0,sizeof(pre));
    for(i=0;i<n;i++){
        for(j=0;j<m;j++){
            scanf("%d",&map[i][j]);
            if(map[i][j]==0)
                st[i][j]=(1<<(k++));
        }
    }
    K=(1<<k);
    for(i=0;i<n;i++)
        for(j=0;j<m;j++){
            for(p=0;p<K;p++)
                dp[i][j][p]=INF;
            if(st[i][j])
                dp[i][j][st[i][j]]=0;
        }
}
void SPFA(){
    int i;
    while(!q.empty()){
        int x=q.front()/1000000,y=(q.front()-x*1000000)/10000,z=q.front()%10000;
        vis[x][y][z]=0;
        q.pop();
        for(i=0;i<4;i++){
            int tx=x+h[i],ty=y+g[i],ts=st[tx][ty]|z;
            if(tx<0 || tx>=n || ty<0 || ty>=m)continue;
            if(dp[tx][ty][ts]>dp[x][y][z]+map[tx][ty]){
                dp[tx][ty][ts]=dp[x][y][z]+map[tx][ty];
                pre[tx][ty][ts]=x*1000000+y*10000+z;
                if(!vis[tx][ty][ts] && ts==z){
                    vis[tx][ty][ts]=1;
                    q.push(tx*1000000+ty*10000+ts);
                }
            }
        }
    }
}
void Steiner_Tree(){
    int i,j,p,x;
    for(p=0;p<K;p++){
        for(i=0;i<n;i++)
            for(j=0;j<m;j++){
		if(st[i][j]&&!(st[i][j]&p)) continue;   //没有这一句WA了一晚上
                for(x=(p-1)&p;x;x=(x-1)&p){
                    if(dp[i][j][p]>dp[i][j][x|st[i][j]]+dp[i][j][(p-x)|st[i][j]]-map[i][j]){
                        dp[i][j][p]=dp[i][j][x|st[i][j]]+dp[i][j][(p-x)|st[i][j]]-map[i][j];
                        pre[i][j][p]=i*1000000+j*10000+(x|st[i][j]);
                    }
                }
                if(dp[i][j][p]!=INF){
                    q.push(i*1000000+j*10000+p);
                    vis[i][j][p]=1;
                }
            }
        SPFA();
    }
}
void dfs(int x,int y,int z){
    int i,j,k;
    ok[x][y]=1;
    int t=pre[x][y][z],tx,ty,ts;
    if(t==0)return;
    tx=t/1000000;
    ty=(t-tx*1000000)/10000;
    ts=t%10000;
    dfs(tx,ty,ts);
    if(x==tx&&y==ty) dfs(x,y,(z-ts)|st[x][y]);
}
void print(){
    int i,j,p,q;
    for(i=0;i<n;i++)
        for(j=0;j<m;j++){
            if(st[i][j]==0)continue;
            printf("%d\n",dp[i][j][K-1]);
            memset(ok,0,sizeof(ok));
            dfs(i,j,K-1);
            for(p=0;p<n;p++){
                for(q=0;q<m;q++){
                    if(st[p][q])printf("%c",'x');
                    else if(ok[p][q]) printf("%c",'o');
                    else printf("%c",'_');
                }
                printf("\n");
            }
            return ;
        }
}
int main(){
    int i,j;
    scanf("%d %d",&n,&m);
    init();
    Steiner_Tree();
    print();
    return 0;
}