poj 3026 Borg Maze 最小生成树+bfs

 

poj 3026 Borg Maze

http://poj.org/problem?id=3026

 

从S点开始，可以出发任意多人开始走，走的相同的路只算一个人的路程，人只能在A点与S点处分开，求S点开始遍历所有A点所经过的最小路程

 

注意m与n后可能有空格，不能用getchar，要用gets………………

 

 

#include <iostream>#include <cstdio>#include <cstring>#include <queue>using namespace std;const int INF= 0x3f3f3f3f;char mit[60][60];int n, m, tot, vis[60][60], dis[110][110];int mp[110];int mov[4][2]={1, 0, 0, 1, -1, 0, 0, -1};int link[110], visc[110];struct node{    int x, y, d;};void init(){    int i, j, k;    scanf("%d%d", &n, &m);    gets(mit[0]);    for( i=0; i<m; i++)        gets(mit[i]);    memset( dis, 0, sizeof( dis));    memset( mp, 0, sizeof( mp));    for( i=1, tot=1; i<m-1; i++){        for( j=1; j<n-1; j++)            if( mit[i][j] == 'S' || mit[i][j]== 'A'){                mp[(i-1)*(n-2)+j]= tot++;                //cout<<(i-1)*(n-2)+j<<"  "<<tot<<endl;            }    }}bool check(int x, int y){    if( x>=1 && x<m && y>=1 && y<n && mit[x][y]!= '#' && !vis[x][y])        return true;    else return false;}void bfs(int sx, int sy){    int i, j, k;    node start, head, next;    queue <node > q;    memset( vis, 0, sizeof( vis));    start.x= sx;    start.y= sy;    start.d= 0;    q.push(start);    vis[start.x][start.y]= 1;    while( !q.empty()){        head= q.front();        q.pop();        if( mit[head.x][head.y] == 'A' || mit[head.x][head.y]== 'S'){            dis[mp[(head.x-1)*(n-2) + head.y]][mp[(sx-1)*(n-2)+ sy]]= head.d;          //  cout<<mp[(head.x-1)*(n-2) + head.y]<<"  "<<mp[(sx-1)*(n-2)+ sy]<<"  "<<head.d<<"   ddd"<<endl;        }        next.d= head.d+1;        for( i=0; i<4; i++){            next.x= head.x+ mov[i][0];            next.y= head.y+ mov[i][1];            if( check(next.x , next.y)){                vis[next.x][next.y]= 1;                q.push(next);            }        }    }}int road(){    int i, j, k, t, tmp, num, cnt= (m-2)*(n-2);    memset( link, 0, sizeof(link));    tot--;    for( i=1; i<=tot; i++){        link[i]= dis[1][i];       // cout<<link[i]<<"  ";    }    link[1]= 0;    visc[1]= 1;    link[1]= 0;    num= 0;    for( i=1; i<tot; i++){        t= INF;        for( j=1; j<=tot; j++){            if( !visc[j] && link[j]< t){                tmp= j;                t= link[j];            }        }        visc[tmp]= 1;        num+= t;        for( j=1; j<=tot; j++){            if( !visc[j] && link[j] > dis[tmp][j])                link[j]= dis[tmp][j];        }    }    return num;}int main(){   // freopen("1.txt", "r", stdin);    int i, j, k, T;    scanf("%d", &T);    while( T--){        init();        for( i=1; i<m-1; i++){            for( j=1; j<n-1; j++)                if( mit[i][j] == 'S' || mit[i][j]== 'A'){                    bfs( i, j);                }        }        memset( visc, 0, sizeof( visc));        printf("%d\n", road());    }    return 0;}