首页 > 代码库 > NYOJ 284 坦克大战 【BFS】+【优先队列】

NYOJ 284 坦克大战 【BFS】+【优先队列】

坦克大战

时间限制:1000 ms  |  内存限制:65535 KB
难度:3
描述
Many of us had played the game "Battle city" in our childhood, and some people (like me) even often play it on computer now. 
What we are discussing is a simple edition of this game. Given a map that consists of empty spaces, rivers, steel walls and brick walls only. Your task is to get a bonus as soon as possible suppose that no enemies will disturb you (See the following picture). 

Your tank can‘t move through rivers or walls, but it can destroy brick walls by shooting. A brick wall will be turned into empty spaces when you hit it, however, if your shot hit a steel wall, there will be no damage to the wall. In each of your turns, you can choose to move to a neighboring (4 directions, not 8) empty space, or shoot in one of the four directions without a move. The shot will go ahead in that direction, until it go out of the map or hit a wall. If the shot hits a brick wall, the wall will disappear (i.e., in this turn). Well, given the description of a map, the positions of your tank and the target, how many turns will you take at least to arrive there?
输入
The input consists of several test cases. The first line of each test case contains two integers M and N (2 <= M, N <= 300). Each of the following M lines contains N uppercase letters, each of which is one of ‘Y‘ (you), ‘T‘ (target), ‘S‘ (steel wall), ‘B‘ (brick wall), ‘R‘ (river) and ‘E‘ (empty space). Both ‘Y‘ and ‘T‘ appear only once. A test case of M = N = 0 indicates the end of input, and should not be processed.
输出
For each test case, please output the turns you take at least in a separate line. If you can‘t arrive at the target, output "-1" instead.
样例输入
3 4
YBEB
EERE
SSTE
0 0
样例输出
8

开始用DFS,结果超时,然后百度的结果是要用优先队列,之前从没见过这玩意儿,然后折腾了一夜晚,第二天才写出来。

#include <cstdio>
#include <cstring>
#include <queue>
using std::priority_queue;
int m, n;
char map[302][302];
bool vis[302][302];
struct Node{
	int x, y, steps;
	friend bool operator<(Node a, Node b){
		return a.steps > b.steps;
	}
} you, tar;
int mov[][2] = {0, 1, 0, -1, 1, 0, -1, 0};
priority_queue<Node> PQ;

int check(Node a){
	if(a.x < 0 || a.y < 0 || a.x >= m || a.y >= n)
		return 0;
	if(vis[a.x][a.y]) return 0;
	if(map[a.x][a.y] == 'B') return 2;
	if(map[a.x][a.y] == 'E') return 1;
	if(map[a.x][a.y] == 'T') return 1;
	return 0;
}

int BFS(){
	Node temp, sta;
	int count;
	vis[you.x][you.y] = 1;
	PQ.push(you);
	while(!PQ.empty()){
		sta = temp = PQ.top(); PQ.pop();
		for(int i = 0; i < 4; ++i){
			temp.x += mov[i][0];
			temp.y += mov[i][1];
			if(count = check(temp)){
				temp.steps += count;
				if(map[temp.x][temp.y] == 'T') 
					return temp.steps;
				vis[temp.x][temp.y] = 1;
				PQ.push(temp);
			}
			temp = sta;
		}
	}
	return -1;
}

int main(){
	while(scanf("%d%d", &m, &n), m || n){
		for(int i = 0; i < m; ++i){
			scanf("%s", map[i]);
			for(int j = 0; j < n; ++j)
				if(map[i][j] == 'Y') you.x = i, you.y = j;
				else if(map[i][j] == 'T') tar.x = i, tar.y = j;
		}
		memset(vis, 0, sizeof(vis));
		while(!PQ.empty()) PQ.pop();
		printf("%d\n", BFS());
	}
	return 0;
}