先bfs 再prim
看的kuangbin巨巨的解题报告
首先move会和c++的move发生冲突 ,编译器显示为ambigious;
其次学会用pair
vector<pair<int,int> >p;
要用空格
make_pair pair.first pair.second
代码如下
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
char s[55][55];
bool used[105];
int mincost[105];
int cost[105][105],a[55][55];
const int INF=0x3f3f3f;
int m[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
int tol,x,y;
int t[55][55];
void bfs(int sx,int sy)
{
queue<pair<int,int> >q;
while(!q.empty()) q.pop();
memset(t,-1,sizeof(t));
t[sx][sy]=0;
q.push(make_pair(sx,sy));
while(!q.empty()){
pair<int,int> now=q.front();
q.pop();
if(a[now.first][now.second]!=-1)
cost[a[sx][sy]][a[now.first][now.second]]=t[now.first][now.second];
for(int i=0;i<4;i++){
int tx=now.first+m[i][0];
int ty=now.second+m[i][1];
if(s[tx][ty]=='#'||t[tx][ty]!=-1) continue;
t[tx][ty]=t[now.first][now.second]+1;
q.push(make_pair(tx,ty));
}
}
}
int prim()
{
for(int i=0;i<tol;i++)
mincost[i]=INF;
int res=0;
memset(used,false,sizeof(used));
mincost[0]=0;
for(int i=1;i<=tol;i++){
int v=-1;
for(int u=0;u<tol;u++)
if(!used[u]&&(v==-1||mincost[u]<mincost[v])) v=u;
if(v==-1) break;
res+=mincost[v];
used[v]=true;
for(int u=0;u<tol;u++)
mincost[u]=min(mincost[u],cost[v][u]);
}
return res;
}
int main()
{
int t;
scanf("%d",&t);
while(t--){
scanf("%d%d",&y,&x);
gets(s[0]);
memset(a,-1,sizeof(a));
tol=0;
for(int i=0;i<x;i++){
gets(s[i]);
for(int j=0;j<y;j++)
if(s[i][j]=='A'||s[i][j]=='S')
a[i][j]=tol++;
}
for(int i=0;i<x;i++)
for(int j=0;j<y;j++)
if(a[i][j]!=-1)
bfs(i,j);
printf("%d\n",prim());
}
}