链接:
题意:
一个M X N 的迷宫里有1个起点和X个目标点,求起点到所有目标点的总步数;搜索迷宫的代价被定义成参与搜索的所有子群体走过的步数总和。例如,如果原群体走了5 步,然后分裂成2 个子群体,每个子群体又分别走了3 步,则总步数为:11=5+3+3。
题解:
通过BFS 起点和目标点 建边,再用prim算法即可答案
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define maxx 333
struct node
{
int x,y;
} head;
char map[maxx][maxx];
int edge[maxx][maxx];
int lowdis[maxx];
int num[maxx][maxx]; //存点的下标
int vis[maxx][maxx],dis[maxx][maxx];
int dir[4][2]= {1,0,-1,0,0,1,0,-1};
int m,n,s,ans;
void bfs(int sx,int sy)
{
queue<node>q;
memset(vis,0,sizeof(vis));
vis[sx][sy]=1;
dis[sx][sy]=0;
q.push((node){sx,sy});
while(!q.empty())
{
head=q.front();
q.pop();
if(map[head.x][head.y]=='S'||map[head.x][head.y]=='A')
edge[num[sx][sy]][num[head.x][head.y]]=dis[head.x][head.y];
int tx,ty;
for(int i=0; i<4; i++)
{
tx=head.x+dir[i][0];
ty=head.y+dir[i][1];
if(tx<0||ty<0||tx>=m||ty>=n||map[tx][ty]=='#'||vis[tx][ty])
continue;
vis[tx][ty]=1;
dis[tx][ty]=dis[head.x][head.y]+1;
q.push((node){tx,ty});
}
}
return;
}
void prim()
{
int i,j,loc,minn;
for(i=0;i<=s;i++)
lowdis[i]=edge[0][i];
lowdis[0]=-1;
for(i=1;i<s;i++)
{
minn=9999999;
for(j=0;j<s;j++)
if(lowdis[j]!=-1&&lowdis[j]<minn)
minn=lowdis[j],loc=j;
ans+=lowdis[loc];
lowdis[loc]=-1;
for(j=0;j<s;j++)
if(lowdis[j]>edge[loc][j])
lowdis[j]=edge[loc][j];
}
return ;
}
int main()
{
int t,i,j;
scanf("%d",&t);
while(t--)
{
s=ans=0;
memset(num,0,sizeof(num));
scanf("%d%d",&n,&m);
gets(map[m]);
for(i=0; i<m; i++)
gets(map[i]);
for(i=0; i<m; i++)
for(j=0; j<n; j++)
if(map[i][j]=='A'||map[i][j]=='S')
num[i][j]=s++;
for(i=0; i<m; i++)
for(j=0; j<n; j++)
{
if(map[i][j]=='S'||map[i][j]=='A')
bfs(i,j);
if(i==j)
edge[i][j]=999999;
}
prim();
cout<<ans<<endl;
}
return 0;
}