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;
}