UVa 11624 Fire

UVa 11624 Fire

題目鏈接 洛谷也有
題目大意：
Joe在迷宮中工作。不幸的是，迷宮的一部分着火了，請幫助喬逃離迷宮。根據喬在迷宮中的位置以及迷宮的哪個方塊着火，你必須確定火焰燒到他之前，喬是否可以離開迷宮，如果能離開他能跑多快。Joe和火每分鐘移動一個方格，上、下、左、右，四個方向中的一個。火勢向四個方向同時蔓延。Joe可以從迷宮的任何一個邊界逃離迷宮。無論是喬還是火都不會到達有牆的位置。
注意：可能會有多個着火點，即多個F

題目思路：
將Fire和Joe分開進行BFS,
然後根據二者到達maze每個邊緣square的時間大小判斷是否能逃生，以及最短逃生時間
Joe是基本的BFS
而Fire是多源點的BFS

具體代碼：

#include<iostream>
#include<queue>
#include<cstdio>
#include<string>

using namespace std;

int step[4][2] = { {1,0},{-1,0},{0,1},{0,-1} };
int maze[1005][1005];
int cost_by_F[1005][1005]; //表示火蔓延到該處的時間
int cost_by_J[1005][1005];	//表示joe跑到該處的時間
int visit[1005][1005];	//表示square是否着火
const int INF = 1e5;
int ans =INF;
int r, c;

void cmp_time()
{
	for (int i = 0; i < r; i++)
		for (int j = 0; j < c; j++)
		{
			//處於maze邊緣，而且可到達
			if (i == 0 || j == 0 || i == r - 1 || j == c - 1 && maze[i][j]) {
				if (cost_by_J[i][j] < cost_by_F[i][j])	//比火先到達，可逃生
				{
					if (cost_by_J[i][j] < ans) {	//更新逃生最小時間
						ans = cost_by_J[i][j];
					}
				}
			}
		}
}
void bfs(int x,int y,int cost[][1005],int choice)
{
	for (int i = 0; i < r; i++)
		for (int j = 0; j < c; j++)
			cost[i][j] = INF, visit[i][j] = 0;//初始化
	queue<pair<int, int>> q;
	if (choice == 1) {	//J的bfs代碼塊
		q.push(pair<int, int>(x, y));
		visit[x][y] = 1;
		cost[x][y] = 1;
	}
	else {	//F的bfs代碼塊
		for (int i = 0; i < r; i++)
			for (int j = 0; j < c; j++)
				if (maze[i][j] == 2) {
					q.push(pair<int, int>(i, j));
					visit[i][j] = 1;
					cost[i][j] = 1;
				}
	}
	while (!q.empty())
	{
		pair<int, int> t = q.front();
		q.pop();		
		for (int i = 0; i < 4; i++)
		{
			int x = t.first, y = t.second;
			x += step[i][0], y += step[i][1];
			if (x >= 0 && x < r &&y >= 0 && y < c && !visit[x][y] && maze[x][y]) {
				visit[x][y] = 1;
				cost[x][y] = cost[t.first][t.second] + 1;
				q.push(pair<int, int>(x, y));
			}
		}
	}

}
int main()
{
	int n;
	cin >> n;
	for (int k = 1; k <= n;k++)
	{
		cin >> r >> c;
		int x, y;
		for (int i = 0; i < r; i++)
		{
			string s;
			cin >> s;
			for (int j = 0; j < c; j++)
			{
				if(s[j]=='#')maze[i][j] = 0;
				else if (s[j]=='.')maze[i][j] = 1;
				else if (s[j] == 'F') {
					maze[i][j] = 2;	//2代表Fire的位置		
				}
				else {
					maze[i][j] = 1;	
					x = i, y = j;	//記錄Joe初始位置
				}
			}	
		}	
		bfs(x, y, cost_by_J, 1);
		bfs(x, y, cost_by_F, 2);
		cmp_time();
		if (ans != INF)printf("%d\n", ans);
		else printf("IMPOSSIBLE\n", k);
		ans = INF;
	}  
	return 0;
}