POJ-3026 Borg Maze （BFS+最小生成树）

题目链接：点击打开链接

题意：

给你一个地图，地图中A代表外星人，S代表出发点，从S点出发，每遇到一个A点便可分裂成多个。

求把所有A都吃完需要多少步。

刚开始没读好题，我以为S点随时都能分裂，头疼了好一会。

思路：

其实A和S是一样的，本题的本质就是求连接A和S的最小生成树。无非麻烦的一点就是各个结点之间的权值。

这里采取的办法是，枚举每一个结点进行一遍BFS，求该结点到其他所有结点的距离，若距离小于初始值，则更新该距离。

这样最终得到的就是每两个结点之间的最短距离。

代码：

#include <iostream>
#include <string>
#include <string.h>
#include <queue>
#include<stdio.h>
using namespace std;

const int inf = 1 << 20;

string *map;
int dis[110][110], x, y, alien[51][51], edge[110][110], num;
bool vis[51][51];
int Move[][2] = {0,-1,0,1,-1,0,1,0};

void bfs(int sx, int sy)
{
	queue<pair<int,int> >q;
	pair<int,int>p;
	p.first = sx, p.second = sy;
	q.push(p);
	memset(vis,false,sizeof(vis));
	vis[sx][sy] = true;
	memset(dis,0,sizeof(dis));
	while(!q.empty())
	{
		int xx = q.front().first;
		int yy = q.front().second;
		q.pop();
		if(alien[xx][yy])
			edge[ alien[sx][sy] ][ alien[xx][yy] ] = dis[xx][yy];
		for(int i = 0;i < 4;i++)
		{
			int mx = xx + Move[i][0];
			int my = yy + Move[i][1];
			if(mx<0||my<0||mx>=y||my>=x)
				continue;
			if(vis[mx][my] || map[mx][my] == '#')
				continue;
			vis[mx][my]  = true;
			dis[mx][my] = dis[xx][yy] + 1;
			p.first = mx, p.second = my;
			q.push(p);
		}
	}
}

int Prim()
{
	int Dis[110];
	bool book[110];
	book[1] = true;
	int t = 1, point, sum = 0, Min;
	for(int i = 1;i <= num;i++)
		Dis[i] = inf, book[i] = false;
	for(int k = 2;k <= num;k++)
	{
		Min = inf;
		for(int i = 2;i <= num;i++)
		{
			if(!book[i] && Dis[i] > edge[t][i])
				Dis[i] = edge[t][i];
			if(!book[i] && Dis[i] < Min)
			{
				Min = Dis[i];
				point = i;
			}
		}
		t = point;
		sum += Min;
		book[t] = true;
	}
	return sum;
}

int main()
{
    int N;
    cin>>N;
    while(N--)
	{
		memset(alien,0,sizeof(alien));
		num = 0;
		cin>>x>>y;
		string a;
		getline(cin,a);

		map = new string[y];
		for(int i = 0;i < y;i++)
		{
			getline(cin,map[i]);
			for(int j = 0;j < x;j++)
				if(map[i][j]=='S'||map[i][j] == 'A')
					alien[i][j] = ++num;
		}

		for(int i = 0;i < y;i++)
			for(int j = 0;j < x;j++)
				if(alien[i][j])
					bfs(i,j);
		cout<<Prim()<<endl;
	}
    return 0;
}