POJ 3026 (BFS+最小生成树)

题意好难理解~~

大意：在一幅图里，‘S’代表起点，‘A’代表外星人，‘#’代表墙，‘  ’代表通路。你带着一群人从S出发，去找到所有的A，每到一个A处，你的队伍可以分为很多支，继续从该点开始遍历，要求找到所有的外星人，并且路径长度最短。最小生成树问题，A和S可以视为相同的。首先需要将图上的点之间的距离计算出来，BFS即可解决，然后在kruskal。累累累~

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
using namespace std;
#define maxx 105
int k,edsum,p[maxx];
int sx,sy,gx,gy;
int m,n;
int dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
char s[53][53];
int vis[53][53],flag[53][53];
 struct Edge
{
  int u,v,w;
}edge[maxx*maxx];
struct Pos
{
  int xx,yy;
}pos[maxx];
struct node
{
  int x,y,step;
};
bool cmp(Edge a,Edge b)
{
  return a.w<b.w;
}
int findpa(int x)
{
  return p[x]==x?x:p[x]=findpa(p[x]);
}
int kruskal()
{
  for(int i=0;i<k;i++)
  p[i]=i;
  int ans = 0, num = 0;
  sort(edge,edge+edsum,cmp);
  for(int i=0;i<edsum;i++)
  {
    int p1=findpa(edge[i].u);
    int p2=findpa(edge[i].v);
    if(p1!=p2)
	{
	  ans+=edge[i].w;
      num++;
	  p[p1]=p2;
	}
    if(num==k-1)
	return ans;
  }
  return ans;
}
void bfs(int p)
{
	int i;
    node a,next;
    queue<node> Q;
    a.x = sx,a.y = sy;
    a.step = 0;
    vis[sx][sy] = 1;
    Q.push(a);
    while(!Q.empty())
    {
        a = Q.front();
        Q.pop();
        if(s[a.x][a.y]=='S'||s[a.x][a.y]=='A')
		{

		  edge[edsum].u=p;
	      edge[edsum].v=flag[a.x][a.y];
	      edge[edsum].w=a.step;
	      edsum++;
        }
        for(i = 0; i<4; i++)
        {

            next.x = a.x+dir[i][0];
            next.y = a.y+dir[i][1];

            if(next.x<0||next.x>=m||next.y<0||next.y>=n||
			vis[next.x][next.y] ==1||s[next.x][next.y]=='#')
			continue;
            vis[next.x][next.y] = 1;
            next.step = a.step+1;
            Q.push(next);
        }
    }
    return;
}
int main()
{
	int t;
    scanf("%d",&t);
    while(t--)
	{
	  k=0;
	  edsum=0;
	  scanf("%d%d",&n,&m);
      gets(s[0]);
	  for(int i=0;i<m;i++)
	  {
	    gets(s[i]);
	    for(int j=0;j<n;j++)
		{
		  if(s[i][j]=='S'||s[i][j]=='A')
		  {
			flag[i][j]=k;
			pos[k].xx=i;
			pos[k++].yy=j;
		  }
		}
	  }
	  for(int i=0;i<k;i++)
	  {
	    memset(vis,0,sizeof(vis));
	    sx=pos[i].xx;
	    sy=pos[i].yy;
	    bfs(i);
	  }
      int ans = kruskal();
	  printf("%d\n",ans);
	 }
	return 0;
}