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

;
bool vis

,vis0[MAXN];
int dis

,pos

,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;
}