POJ 3026 最小生成树 + BFS

题目：

给地图，S为起点，从S开始搜索，遇到A后就把A同化为S，与之前的S一同继续传播。这就可以看出是最小生成树的特征。

难点在于，如何构建这个图。

需要计算任意两个点之间的最短城市距离（这个因为有墙，不能穿墙），可以通过BFS求，

读入数据的时候把A或S点标记为图的节点，然后对每个节点进行BFS，构建当前节点到其他节点的边。

这样图就建成了，剩下就是最小生成树了。

 

Code:

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int N = 105;
typedef pair<int,int> pii;
int e[N][N];
int d[N];
bool vis[N];
bool mark[N][N];
int alien[N][N];
int dis[N][N];
int n,x,y;;
string *mp;

int dir[4][2] = {{-1,0},{0,1},{1,0},{0,-1}};

void bfs(int sx,int sy){
	queue<pii> q;
	memset(mark,0,sizeof(mark));
	memset(dis,0,sizeof(dis));
	q.push(pii(sx,sy));
	mark[sx][sy] = 1;
	while(!q.empty()){
		pii cur = q.front(); q.pop();
		int xx = cur.first;
		int yy = cur.second;
		if(alien[xx][yy]){//说明这个是个顶点 
			e[alien[sx][sy]][alien[xx][yy]] = dis[xx][yy];
		}
		for(int i = 0; i < 4; ++i){
			int nx = xx + dir[i][0];
			int ny = yy + dir[i][1];
			if(nx < 0 || nx >= x || ny < 0 || ny >= y || mp[nx][ny] == '#')
				continue;
			if(mark[nx][ny]) continue;
			 dis[nx][ny] = dis[xx][yy] + 1;
			 mark[nx][ny] = 1;
			 q.push(pii(nx,ny));
		}
	}
}

int prim(int s,int n){
	for(int i = 1;i <= n; ++i) d[i] = (i == s ? 0 : e[s][i]),vis[i] = 0;
	vis[s] = 1;
	int ans = 0;
	while(1){
		int u = -1;
		for(int v = 1; v <= n; ++v){
			if(!vis[v] && (u == -1 || d[v] < d[u])) u = v;
		}
		if(u == -1) break;
		vis[u] = true;
		ans += d[u];
		for(int v = 1; v <= n; ++v){
			if(!vis[v] && d[v] > e[u][v]){
				d[v] = e[u][v];
			}
		}
	}
	return ans;
}

int main()
	{
		ios::sync_with_stdio(false);
		cin.tie(0);
		int t,num;
		string str;
		cin >> t;	
		while(t--){
			cin >> y >> x;
			memset(e,0x3f3f,sizeof(e));
			memset(alien,0,sizeof(alien));
			mp = new string[x];
			num = 0;
			getline(cin,str);
			for(int i = 0; i < x; ++i){
				getline(cin,mp[i]);
				for(int j = 0; j < y; ++j){
					if(mp[i][j] == 'A' || mp[i][j] == 'S'){
						alien[i][j] = ++num;
					}
				}
			}
			for(int i = 0; i < x; ++i){
				for(int j = 0; j < y; ++j){
					if(alien[i][j]){
						bfs(i,j);
					}
				}
			}
			int ans = prim(1,num);
			cout << ans << endl;
		}
		
		return 0;
	}