Borg Maze poj 3026

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
char map[55][55];
int n,m;
int go[4][2]={1,0,0,1,-1,0,0,-1};
int f[2500];
struct node//用于记录任意两个标记点之间最短距离
{
int u,v,w;
}q[100000];
struct node1{
int r,c;
int step;
}p[2500];
int tmp[55][55];//该数组用于记录bfs的起始点到图中任意点的最短距离
bool cmp(node a,node b)
{
return a.w<b.w;
}
int getf(int k)
{
return k==f[k]?k:f[k]=getf(f[k]);
}
int bfs(int s)
{
int book[55][55];
memset(book,0,sizeof(book));
memset(tmp,0,sizeof(tmp));
queue<node1> que;
que.push(node1{p[s].r,p[s].c,0});
while(!que.empty())
{
node1 t=que.front(),a;
que.pop();
for(int i=0;i<4;i++)
{
a=t;
a.r+=go[i][0];
a.c+=go[i][1];
a.step++;
if(!book[a.r][a.c]&&0<=a.r&&a.r<m&&0<=a.c&&a.c<n&&map[a.r][a.c]!='#')
{
book[a.r][a.c]=1;
tmp[a.r][a.c]=a.step;
que.push(a);
}
}
}
}
int main()
{
int t;
char c;
scanf("%d",&t);
while(t--)
{
memset(q,0,sizeof(q));
memset(p,0,sizeof(p));
memset(map,0,sizeof(map));
int cnt=1;
scanf("%d%d",&n,&m);
c=getchar();
while(c!='\n')c=getchar();
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
scanf("%c",&map[i][j]);
if(map[i][j]=='A'||map[i][j]=='S')//给每个标记点标号
{
p[cnt].r=i;
p[cnt++].c=j;
}
}
getchar();
}
int t=0,k=0;
for(int x=1;x<cnt;x++)
{
k=0;
bfs(x);
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
if(map[i][j]=='A'||map[i][j]=='S')//将标号为x的标记点到任意标记点的最短距离记录下来
{
k++;
if(k!=x)
{
q[t].u=x;
q[t].v=k;
q[t++].w=tmp[i][j];
}
}
}
}
}
sort(q,q+t,cmp);
for(int i=0;i<=cnt;i++)
f[i]=i;
int count=1,i=0,ans=0;
while(count<cnt-1)//kruskal
{
if(getf(q[i].u)!=getf(q[i].v))
{
f[getf(q[i].u)]=getf(q[i].v);
count++;
ans+=q[i].w;
//printf("%d  %d   %d\n",q[i].u,q[i].v,q[i].w);
}
i++;
}
printf("%d\n",ans);
}
return 0;
}