POJ 3026 Brog Maze BFS+最小生成树

链接：

3026

题意：

一个M X N 的迷宫里有1个起点和X个目标点，求起点到所有目标点的总步数；搜索迷宫的代价被定义成参与搜索的所有子群体走过的步数总和。例如，如果原群体走了5 步，然后分裂成2 个子群体，每个子群体又分别走了3 步，则总步数为：11=5+3+3。

题解：

通过BFS 起点和目标点 建边,再用prim算法即可答案

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define maxx 333
struct node
{
    int x,y;
} head;
char map[maxx][maxx];
int edge[maxx][maxx];
int lowdis[maxx];
int num[maxx][maxx]; //存点的下标
int vis[maxx][maxx],dis[maxx][maxx];
int dir[4][2]= {1,0,-1,0,0,1,0,-1};
int m,n,s,ans;
void bfs(int sx,int sy)
{
    queue<node>q;
    memset(vis,0,sizeof(vis));
    vis[sx][sy]=1;
    dis[sx][sy]=0;
    q.push((node){sx,sy});
    while(!q.empty())
    {
        head=q.front();
        q.pop();
        if(map[head.x][head.y]=='S'||map[head.x][head.y]=='A')
               edge[num[sx][sy]][num[head.x][head.y]]=dis[head.x][head.y];
        int tx,ty;
        for(int i=0; i<4; i++)
        {
            tx=head.x+dir[i][0];
            ty=head.y+dir[i][1];
            if(tx<0||ty<0||tx>=m||ty>=n||map[tx][ty]=='#'||vis[tx][ty])
                continue;
            vis[tx][ty]=1;
            dis[tx][ty]=dis[head.x][head.y]+1;
            q.push((node){tx,ty});
        }
    }
    return;
}
void prim()
{
    int i,j,loc,minn;
    for(i=0;i<=s;i++)
        lowdis[i]=edge[0][i];
    lowdis[0]=-1;
    for(i=1;i<s;i++)
    {
        minn=9999999;
        for(j=0;j<s;j++)
            if(lowdis[j]!=-1&&lowdis[j]<minn)
            minn=lowdis[j],loc=j;
        ans+=lowdis[loc];
        lowdis[loc]=-1;
        for(j=0;j<s;j++)
            if(lowdis[j]>edge[loc][j])
            lowdis[j]=edge[loc][j];
    }
    return ;
}
int main()
{
    int t,i,j;
    scanf("%d",&t);
    while(t--)
    {
        s=ans=0;
        memset(num,0,sizeof(num));
        scanf("%d%d",&n,&m);
        gets(map[m]);
        for(i=0; i<m; i++)
            gets(map[i]);
        for(i=0; i<m; i++)
            for(j=0; j<n; j++)
                if(map[i][j]=='A'||map[i][j]=='S')
                    num[i][j]=s++;
        for(i=0; i<m; i++)
            for(j=0; j<n; j++)
               {
                   if(map[i][j]=='S'||map[i][j]=='A')
                   bfs(i,j);
                   if(i==j)
                    edge[i][j]=999999;
               }
       prim();
       cout<<ans<<endl;
    }
    return 0;
}