hdu1429 胜利大逃亡(续)--BFS

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1429

一：分析

定义一个三维数组，标记该点是否走过，其中第三维代表在该路径上所获钥匙的标记。

二：AC代码

#define _CRT_SECURE_NO_DEPRECATE 
#define _CRT_SECURE_CPP_OVERLOAD_STANDARD_NAMES 1 

#include<iostream>
#include<algorithm>
#include<queue>
#include<cctype>
using namespace std;

struct Node
{
	int x, y;
	int step;
	int state;
	Node(){}
	Node(int x, int y, int step, int state) :x(x), y(y), step(step), state(state) {}
};

int dir[4][2] = { { 0, 1 },{ 0, -1 },{ 1, 0 },{ -1, 0 } };
char s[25][25];
bool vis[25][25][1 << 10];
int n, m, t;

int get(char c)
{
	return c - (isupper(c) ? 'A' : 'a');
}

bool check(int x, int y)
{
	if (x < 0 || x >= n || y < 0 || y >= m || s[x][y] == '*')
		return false;
	return true;
}


int bfs(int x, int y)
{
	queue<Node> Q;
	vis[x][y][0] = 1;

	Q.push(Node(x, y, 0, 0));

	while (!Q.empty())
	{
		Node cur = Q.front();
		Q.pop();
		if (cur.step >= t - 1)
			break;

		for (int i = 0; i < 4; i++)
		{
			Node temp = cur;
			temp.x += dir[i][0];
			temp.y += dir[i][1];
			temp.step++;

			if (check(temp.x, temp.y) == 0 || vis[temp.x][temp.y][temp.state])
				continue;

			if (isupper(s[temp.x][temp.y]) && !(temp.state&(1 << get(s[temp.x][temp.y]))))//遇到门，没有钥匙，不走
				continue;

			if (islower(s[temp.x][temp.y]))//找到钥匙
				temp.state |= (1 << get(s[temp.x][temp.y]));
               
			if (s[temp.x][temp.y] == '^')
				return temp.step;

			if (!vis[temp.x][temp.y][temp.state])
			{
				vis[temp.x][temp.y][temp.state] = 1;
				Q.push(temp);
			}
		}
	}

	return -1;
}

int main()
{
	int x, y;
	while (~scanf("%d%d%d", &n, &m, &t))
	{
		for (int i = 0; i < n; i++)
		{
			getchar();
			for (int j = 0; j < m; j++)
			{
				scanf("%c", &s[i][j]);
				if (s[i][j] == '@')
				{
					x = i;
					y = j;
				}
			}
		}

		memset(vis, 0, sizeof(vis));

		printf("%d\n", bfs(x, y));
	}

	return 0;
}