牛客网-拼多多2018笔试 迷宫寻路(状态压缩+bfs)

题目来源:迷宫寻路

题目描述

假设一个探险家被困在了地底的迷宫之中,要从当前位置开始找到一条通往迷宫出口的路径。迷宫可以用一个二维矩阵组成,有的部分是墙,有的部分是路。迷宫之中有的路上还有门,每扇门都在迷宫的某个地方有与之匹配的钥匙,只有先拿到钥匙才能打开门。请设计一个算法,帮助探险家找到脱困的最短路径。如前所述,迷宫是通过一个二维矩阵表示的,每个元素的值的含义如下 0-墙,1-路,2-探险家的起始位置,3-迷宫的出口,大写字母-门,小写字母-对应大写字母所代表的门的钥匙

输入描述:

迷宫的地图,用二维矩阵表示。第一行是表示矩阵的行数和列数M和N 后面的M行是矩阵的数据,每一行对应与矩阵的一行(中间没有空格)。M和N都不超过100, 门不超过10扇。

输出描述:

路径的长度,是一个整数

输入

5 5
02111
01a0A
01003
01001
01111

输出

7

思路

简单状压,利用10个二进制位,遇到一个门,检查一下能不能开门,不能开门就不能走。遇到钥匙的时候用位运算「或」把这个状态添加进去,bfs 搜出来的就是结果。

代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int N = 100 + 10;
int n, m, sx, sy, ex, ey, minn = inf;
char s[N][N];
int vis[N][N][(1 << 10) + 10];
int go[4][2] = {1, 0, -1, 0, 0, 1, 0, -1};
struct node
{
    int x, y, step, key;
    node() {}
    node(int _x, int _y, int _step, int _key)
    {
        x = _x, y = _y, step = _step, key = _key;
    }
};
int pd(int x, int y)
{
    if (x >= 1 && x <= n && y >= 1 && y <= m)
        return 1;
    return 0;
}
int bfs()
{
    node now(sx, sy, 0, 0), to;
    vis[sx][sy][0] = 1;
    queue<node> q;
    q.push(now);
    while (!q.empty())
    {
        now = q.front();
        q.pop();
        if (now.x == ex && now.y == ey)
            return now.step;
        for (int i = 0; i < 4; i++)
        {
            to.x = now.x + go[i][0];
            to.y = now.y + go[i][1];
            to.step = now.step + 1;
            to.key = now.key;
            if (pd(to.x, to.y) && s[to.x][to.y] != '0' && !vis[to.x][to.y][to.key])
            {
                if (s[to.x][to.y] >= 'A' && s[to.x][to.y] <= 'Z')
                {
                    int key = to.key | (1 << (s[to.x][to.y] - 'A'));
                    if (key != to.key)
                        continue;
                }
                if (s[to.x][to.y] >= 'a' && s[to.x][to.y] <= 'z')
                {
                    to.key |= (1 << (s[to.x][to.y] - 'a'));
                }
                vis[to.x][to.y][to.key] = 1;
                q.push(to);
            }
        }
    }
    return 0;
}
int main()
{
    //freopen("in.txt", "r", stdin);
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
    {
        scanf("%s", s[i] + 1);
        for (int j = 1; j <= m; j++)
        {
            if (s[i][j] == '2')
                sx = i, sy = j;
            if (s[i][j] == '3')
                ex = i, ey = j;
        }
    }
    printf("%d\n", bfs());
    return 0;
}

最后修改于 2019-01-19

知识共享许可协议
本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。