POJ3026 - Borg Maze [MST]

题目其实就是给你一个图，求所有字母连起来的最小生成树。字母之间的距离要预处理，可以弗洛伊德也可以bfs。

然后就做MST就好，其实我觉得kru也行但是网上都是用的prim。

代码极丑勿学姿势。。。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
struct dian{
	int x,y;
}d[330],dui[110000];
int t;
int n,m;
string s;
char a[155][155];
int hao[155][155];
int loc;
int e[355][355];
int v[155][155];
int dist[330];
int sett[330];
void BFS(int x,int y){
	memset(v,-1,sizeof(v));
	v[x][y]=0;
	dui[1].x=x;
	dui[1].y=y;
	int head,tail;
	head=0;tail=1;
	int dx[5]={0,-1,0,1,0},dy[5]={0,0,-1,0,1};
	while(head!=tail){
		head++;
		int xx=dui[head].x,yy=dui[head].y;
		for(int i=1;i<=4;i++){
			if(v[xx+dx[i]][yy+dy[i]]==-1&&a[xx+dx[i]][yy+dy[i]]!='#'){
				v[xx+dx[i]][yy+dy[i]]=v[xx][yy]+1;
				e[hao[x][y]][hao[xx+dx[i]][yy+dy[i]]]=v[xx+dx[i]][yy+dy[i]];
				tail++;
				dui[tail].x=xx+dx[i];
				dui[tail].y=yy+dy[i];
			}
		}
	}
}
int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&m,&n);
		getline(cin,s);
		memset(a,'#',sizeof(a));
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++)scanf("%c",&a[i][j]);
			getline(cin,s);
		}
		memset(hao,0,sizeof(hao));
		loc=0;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				if(a[i][j]=='A'||a[i][j]=='S'){
					loc++;
					hao[i][j]=loc;
					d[loc].x=i;
					d[loc].y=j;
				}
			}
		}
//		printf("FXCK\n");
		memset(e,30,sizeof(e));
		for(int i=1;i<=loc;i++)e[i][i]=0;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				if(hao[i][j])BFS(i,j);
			}
		}
		memset(dist,30,sizeof(dist));
		dist[1]=0;
		memset(sett,0,sizeof(sett));
		sett[1]=1;
		for(int i=1;i<=loc;i++)dist[i]=e[1][i];
		int ans=0;
//		printf("FXCK\n");
		for(int k=1;k<loc;k++){
			int maxx=1100000;
			int ji;
			for(int i=1;i<=loc;i++){
				if(!sett[i]){
					if(maxx>dist[i]){
						maxx=dist[i];
						ji=i;
					}
				}
			}
//			printf("ji=%d\n",ji);
			ans+=maxx;
			sett[ji]=1;
			for(int i=1;i<=loc;i++){
				if(!sett[i])dist[i]=min(dist[i],e[ji][i]);
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}