poj 3026（BFS+最小生成树）

题意：给定一张地图，其中包含源点S和A，要求S到所有A点的最短距离的和，另外公共路径不应该算在和中。

 

思路：这是一道明显的求最小生成树的题目。首先求出图中的每个A点和S点到其他A点和S点的最短路径，然后根据求得的路径建边，最后prim。

 

参考代码：

#include <iostream>
#include <cstdio>

using namespace std;

#define MAX 1000000

char str[160][160];
int node[160][160];
int x,y,n;
int edge[160][160],dist[160][160];
struct Move
{
int xx,yy;
};
struct Move move[4]={{-1,0},{0,-1},{1,0},{0,1}};
bool vis[160][160];
int qx[4000],qy[4000],head,tail;

void bfs(int i,int j)
{
int k,l;
memset(vis,0,sizeof(vis));
memset(dist,0,sizeof(dist));
vis[i][j]=true;
head=tail=0;
qx[tail]=i;
qy[tail++]=j;
while(head<tail)
{
int ii=qx[head];
int jj=qy[head++];
if(node[ii][jj])
{
edge[ node[i][j] ][ node[ii][jj] ]=dist[ii][jj];
}
for(k=0;k<4;k++)
{
int xx=ii+move[k].xx;
int yy=jj+move[k].yy;
if(xx>=0&&xx<x&&yy>=0&&yy<y)
{
if(!vis[xx][yy]&&str[xx][yy]!='#')
{
qx[tail]=xx;
qy[tail++]=yy;
vis[xx][yy]=true;
dist[xx][yy]=dist[ii][jj]+1;
}
}
}
}
}

int closedge[160];
bool vis1[160];

int prim()
{
int i,j,k,res=0;
memset(vis1,0,sizeof(vis1));
for(i=1;i<=n;i++)
{
closedge[i]=edge[1][i];
}
vis1[1]=true;
for(j=1;j<=n-1;j++)
{
k=1;
while(vis1[k])k++;
int beg=k;
for(i=k+1;i<=n;i++)
{
if(!vis1[i]&&closedge[i]<closedge[k])
{
k=i;
}
}
vis1[k]=true;
res+=closedge[k];
for(i=1;i<=n;i++)
{
if(!vis1[i]&&edge[k][i]<closedge[i])
{
closedge[i]=edge[k][i];
}
}
}
return res;
}

int main()
{
//	freopen("in.txt","r",stdin);
int i,j,k,t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&y,&x);
char temp[30];
gets(temp);
for(i=0;i<x;i++)
{
gets(str[i]);
}
memset(node,0,sizeof(node));
n=0;
for(i=0;i<x;i++)
{
for(j=0;j<y;j++)
{
if(str[i][j]=='A'||str[i][j]=='S')
{
node[i][j]=++n;
}
}
}
for(i=0;i<x;i++)
{
for(j=0;j<y;j++)
{
if(node[i][j])
{
bfs(i,j);
}
}
}
printf("%d/n",prim());
}
return 0;
}