http://acm.hdu.edu.cn/showproblem.php?pid=5094
现有一个n * m的矩阵。
矩阵相邻两个格子之间可以能有墙或门。
墙是不能通过的,门分为p(p <= 10)种类型,可以被相应类型的钥匙打开然后通过。
各个类型的钥匙分布在各个格子里。
现要求从左上角走到右下角所需花费的最短时间。
典型的状态压缩DP。
状态是当前所处的位置以及拥有哪些钥匙。
BFS并不断根据当前格子所能达到的钥匙更新状态即可。
#include
#include
#include
#include
#include
using namespace std;
const int maxn = 60;
int dx[4] = {-1, 1, 0, 0};
int dy[4] = {0, 0, -1, 1};
int w[maxn][maxn][4];
int dp[maxn][maxn][1 << 10], key[maxn][maxn];
queue, int> > q;
int n, m, p;
inline bool inRange(int x, int y)
{
if (x < 0 || x >= n)
{
return false;
}
if (y < 0 || y >= m)
{
return false;
}
return true;
}
int main()
{
while (scanf("%d %d %d", &n, &m, &p) == 3)
{
memset(w, -1, sizeof(w));
memset(dp, -1, sizeof(dp));
memset(key, 0, sizeof(key));
int k;
scanf("%d", &k);
while (k --)
{
int x1, y1, x2, y2, g;
scanf("%d %d %d %d %d", &x1, &y1, &x2, &y2, &g);
for (int i = 0; i < 4; i ++)
{
if (((x1 + dx[i]) == x2) && ((y1 + dy[i]) == y2))
{
w[x1 - 1][y1 - 1][i] = g;
}
if (((x2 + dx[i]) == x1) && ((y2 + dy[i]) == y1))
{
w[x2 - 1][y2 - 1][i] = g;
}
}
}
int s;
scanf("%d", &s);
while(s --)
{
int x, y, q;
scanf("%d %d %d", &x, &y, &q);
key[x - 1][y - 1] |= (1 << (q - 1));
}
dp[0][0][key[0][0]] = 0;
q.push(make_pair(make_pair(0, 0), key[0][0]));
while(!q.empty())
{
int x = q.front().first.first;
int y = q.front().first.second;
int mask = q.front().second;
q.pop();
for (int i = 0; i < 4; i ++)
{
int nx = x + dx[i];
int ny = y + dy[i];
if (!inRange(nx, ny))
{
continue;
}
int nmask = mask | key[nx][ny];
if (dp[nx][ny][nmask] != -1)
{
continue;
}
if (w[x][y][i] == 0)
{
continue;
}
if ((w[x][y][i] != -1) && (!((nmask >> (w[x][y][i] - 1)) & 1)))
{
continue;
}
dp[nx][ny][nmask] = dp[x][y][mask] + 1;
q.push(make_pair(make_pair(nx, ny), nmask));
}
}
int ans = -1;
for (int mask = 0; mask < (1 << p); mask ++)
{
if (dp[n - 1][m - 1][mask] != -1)
{
ans = (ans == -1) ? dp[n - 1][m - 1][mask]: min(ans, dp[n - 1][m - 1][mask]);
}
}
printf("%d\n", ans);
}
return 0;
}