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 105int 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;}