leetcode.获取所有钥匙的最短路径

最新推荐文章于 2024-10-16 10:13:45 发布

qq_46482117

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量173

点赞数

分类专栏：力扣文章标签： leetcode 算法图论

本文链接：https://blog.csdn.net/qq_46482117/article/details/127794992

版权

力扣专栏收录该内容

21 篇文章 0 订阅

订阅专栏

这篇博客介绍了一种使用宽度优先搜索（BFS）算法解决寻找迷宫中所有钥匙的最短路径问题。作者给出了C++代码实现，通过维护一个状态结构，包括当前位置、钥匙数量，以及利用二维数组记录已访问状态，逐步遍历迷宫并收集钥匙。最终，当所有钥匙都被找到时返回步数，否则返回-1表示无法找到所有钥匙的路径。

摘要由CSDN通过智能技术生成

864. 获取所有钥匙的最短路径

思路
用bfs来做这种迷宫最短路（无权图最短路)

class Solution {
    int dx[4] = {0, 0, -1, 1};
    int dy[4] = {1, -1, 0, 0};
    
    struct State {
        int x;
        int y;
        int key;
    };
    
public:
    int shortestPathAllKeys(vector<string>& grid) {
        int m = grid.size(), n = grid[0].size();
        
        // 先找到起点，以及钥匙的数量
        State start; 
        int keycnt = 0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (grid[i][j] == '@') {
                    start.x = i;
                    start.y = j;
                    start.key = 0;
                }
                if (islower(grid[i][j])) {
                    keycnt++;
                }
            }
        }
        
        // BFS
        queue<State> q;
        q.push(start);
        int allkey = (1 << keycnt) - 1; // 表示所有钥匙都找到的状态
        vector<vector<vector<bool>>> visited(m, vector<vector<bool>>(n, vector<bool>(allkey + 1, false)));
        int step = 0;
        while (!q.empty()) {
            int qSize = q.size();
            for (int i = 0; i < qSize; i++) {
                State cur = q.front();
                q.pop();
                
                // 如果所有钥匙都找到
                if (cur.key == allkey) return step;
                
                // 遍历下一个状态
                for (int k = 0; k < 4; k++) {
                    int x = cur.x + dx[k], y = cur.y + dy[k];
                    if (x < 0 || y < 0 || x >= m || y >= n) continue;
                    
                    char c = grid[x][y];
                    int key = cur.key;
                    
                    if (c == '#') continue;
                    if (isupper(c) && (key >> (c - 'A') & 1) == 0) continue;
                    
                    if (islower(c)) {
                        key |= 1 << (c - 'a');
                    }
                    if (visited[x][y][key] == true) continue;
                    
                    visited[x][y][key] = true; 
                    q.push(State{x, y, key}); 
                } 
            }
            step++;
        }
        return -1;
    }
};