POJ-3026 Borg Maze

题目链接：https://vjudge.net/problem/POJ-3026

把'S'或'A'都看成一个点，先用BFS求出所有点两两之间距离，保存起来

再用最小生成树算法求出答案，输入有坑，输入n和m时需要这样scanf("%d%d\n",&m,&n);

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int N=50+5;
const int MAXN=100+5;
const int inf=0x7f7f7f7f;
const int dx[]={0,0,-1,1};
const int dy[]={-1,1,0,0};
struct node
{
	int x,y;
	node(int a,int b)
	{
		x=a;y=b;
	}
};
char g[N][N];
bool vis[N][N],vis0[MAXN];
int dis[N][N],pos[N][N],g0[MAXN][MAXN],dist[MAXN];
int n,m,cnt;
void bfs(int qx,int qy)
{
	memset(vis,false,sizeof(vis));
	memset(dis,inf,sizeof(dis));
	queue<node> q;
	q.push(node(qx,qy));
	vis[qx][qy]=true;
	dis[qx][qy]=0;
	int ans=0;
	while(!q.empty())
	{
		node u=q.front();
		q.pop();
		int x=u.x;
		int y=u.y;
		if(g[x][y]=='A'||g[x][y]=='S')
		{
			g0[pos[x][y]][pos[qx][qy]]=g0[pos[qx][qy]][pos[x][y]]=dis[x][y];
		}
		for(int i=0;i<4;i++)
		{
			int nx=x+dx[i];
			int ny=y+dy[i];
			if(!vis[nx][ny]&&(g[nx][ny]==' '||g[nx][ny]=='A'||g[nx][ny]=='S'))
			{
				vis[nx][ny]=true;
				dis[nx][ny]=dis[x][y]+1;
				q.push(node(nx,ny));
			}
		}
	}
}
void Prim()
{
	int k,tmp;
	for(int i=0;i<cnt;i++)
	dist[i]=inf;//初始化 
	memset(vis0,false,sizeof(vis0));
	dist[1]=0;
	for(int i=0;i<cnt;i++){
		tmp=inf;
		for(int j=0;j<cnt;j++){
			if(!vis0[j]&&tmp>dist[j]){
				tmp=dist[j];
				k=j;
			}//找出最小距离的节点 
		}
		vis0[k]=true;//把访问的节点做标记
		for(int j=0;j<cnt;j++){
			if(!vis0[j]&&dist[j]>g0[k][j])
			dist[j]=g0[k][j];//更新最短距离 
		}
	}
}
int main()
{
	int T,sx,sy;
	scanf("%d",&T);
	while(T--)
	{
		cnt=0;
		scanf("%d%d\n",&m,&n);
		for(int i=0;i<n;i++)
		for(int j=0;j<=m;j++)
		{
			scanf("%c",&g[i][j]);
			if(g[i][j]=='A'||g[i][j]=='S') pos[i][j]=cnt++;
			if(g[i][j]=='S') sx=i,sy=j;
		}
		memset(g0,inf,sizeof(g0));
		for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
		if(g[i][j]=='A'||g[i][j]=='S') bfs(i,j);
		Prim();
		int ans=0;
		for(int i=0;i<cnt;i++) ans+=dist[i];
		printf("%d\n",ans);
	}
	return 0;
}