J - Borg Maze +getchar 的使用注意(二维字符数组的输入)
题目链接:
https://vjudge.net/contest/66965#problem/J
具体思路:
首先将每个点之间的最短距离求出(bfs),A 或者 S作为起点跑bfs,这样最短距离就求出来了。然后再用最短路的算法求出最小生成树的权值的和就可以了,getchar的注意事项在代码中解释。
#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<stdio.h>
using namespace std;
# define maxn 1000+10
# define inf 0x3f3f3f3f
# define ll long long
int n,m,num;
char a[maxn][maxn];
int vis[maxn][maxn];
int dis[maxn][maxn];
int f[2][4]= {{1,-1,0,0},{0,0,1,-1}};
int p[maxn];
int t[maxn][maxn];
int vis1[maxn*maxn];
struct node
{
int x,y,step;
node() {}
node(int xx,int yy,int zz)
{
x=xx;
y=yy;
step=zz;
}
};
bool judge(int t1,int t2)
{
if(t1>=0&&t1<n&&t2>=0&&t2<m)return true;
return false;
}
void bfs(int t1,int t2)
{
queue<node>q;
while(!q.empty())q.pop();
memset(vis,0,sizeof(vis));
vis[t1][t2]=1;
q.push(node(t1,t2,0));
while(!q.empty())
{
node temp=q.front();
q.pop();
if(t[temp.x][temp.y])
{
dis[t[t1][t2]][t[temp.x][temp.y]]=dis[t[temp.x][temp.y]][t[t1][t2]]=temp.step;
}
for(int i=0; i<4; i++)
{
int x=temp.x+f[0][i];
int y=temp.y+f[1][i];
if(judge(x,y)&&vis[x][y]==0&&a[x][y]!='#')
{
vis[x][y]=1;
q.push(node(x,y,temp.step+1));
}
}
}
}
int prim()
{
memset(vis1,0,sizeof(vis1));
int sum=0;
for(int i=1; i<=num; i++)
{
p[i]=dis[1][i];
}
vis1[1]=1;
for(int j=2; j<=num; j++)
{
int minn=inf,po;
for(int i=1; i<=num; i++)
{
if(!vis1[i]&&p[i]<minn)
{
minn=p[i];
po=i;
}
}
vis1[po]=1;
sum+=minn;
for(int i=1; i<=num; i++)
{
if(!vis1[i]&&dis[po][i]<p[i])
{
p[i]=dis[po][i];
}
}
}
return sum;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
num=0;
memset(t,0,sizeof(t));
scanf("%d %d",&m,&n);
gets(a[0]);//这里要先输入a[0]的原因是二维数组输入的时候会将回车也当成一个字符输入,所以二维数组的最后一行存储不上。
for(int i=0; i<n; i++)
{
gets(a[i]);
for(int j=0; j<m; j++)
{
if(a[i][j]=='S'||a[i][j]=='A')
{
num++;
t[i][j]=num;
}
}
}
for(int i=0; i<n; i++)
{
for(int j=0; j<m; j++)
{
if(t[i][j]==0)continue;
bfs(i,j);
}
}
int ans=prim();
// printf("%d\n",ans);
}
return 0;
}