题目描述:
给定一个二维网格 grid ,其中:
- '.' 代表一个空房间
- '#' 代表一堵
- '@' 是起点
- 小写字母代表钥匙
- 大写字母代表锁
我们从起点开始出发,一次移动是指向四个基本方向之一行走一个单位空间。我们不能在网格外面行走,也无法穿过一堵墙。如果途经一个钥匙,我们就把它捡起来。除非我们手里有对应的钥匙,否则无法通过锁。
假设 k 为 钥匙/锁 的个数,且满足 1 <= k <= 6,字母表中的前 k 个字母在网格中都有自己对应的一个小写和一个大写字母。换言之,每个锁有唯一对应的钥匙,每个钥匙也有唯一对应的锁。另外,代表钥匙和锁的字母互为大小写并按字母顺序排列。
返回获取所有钥匙所需要的移动的最少次数。如果无法获取所有钥匙,返回 -1 。
示例1:
输入:grid = ["@.a.#","###.#","b.A.B"]
输出:8
解释:目标是获得所有钥匙,而不是打开所有锁。
示例2:
输入:grid = ["@..aA","..B#.","....b"]
输出:6
示例3:
输入: grid = ["@Aa"]
输出: -1
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 30
grid[i][j] 只含有 '.', '#', '@', 'a'-'f' 以及 'A'-'F'
钥匙的数目范围是 [1, 6]
每个钥匙都对应一个 不同 的字母
每个钥匙正好打开一个对应的锁
解题思路:
本题是走迷宫类型问题,求起点到终态的最短路。走迷宫和 bfsbfs 思想是一致的,因为 bfsbfs 每次对当前所有状态整体外扩一格,它第一次走到终态,经过的路程一定是最短路。确定使用 bfs,它的起点就是 '@' 所在的坐标,终态是集齐所有钥匙。如何表示终态?可以用状态压缩,让矩阵显示当前持有的钥匙。用每个二进制位代表一把钥匙。二进制位是 1 ,表示有这把钥匙, 0 表示没有这把钥匙。
例如:一共 4把钥匙," 1001 " 表示,当前有第 3 把,第 0 把钥匙,但是没有 2、1 钥匙。
代码:
class Solution {
public:
int shortestPathAllKeys(vector<string>& grid) {
int dx[4] = {-1,0,1,0};//上右下左
int dy[4] = {0,1,0,-1};
int m = grid.size(),n = grid[0].size();//m行n列
int sx = 0,sy =0;//起始地点
int cnt = 0;//统计小写字母(钥匙)数量
for(int i =0;i<m;i++)
for(int j = 0;j<n;j++)
if('@'==grid[i][j]) sx = i,sy = j;
else if(islower(grid[i][j])) cnt++;
queue<tuple<int,int,int>> q({{{sx,sy,0}}});///插入三元组 x,y,k(钥匙)
vector<vector<vector<int>>> dis(m,vector<vector<int>>(n,vector<int>(1<<cnt,-1)));
dis[sx][sy][0] = 0;//初始状态距离为0;
while(q.size()){//队列非空
auto [x,y,k] = q.front();//保存队首坐标,钥匙编码
q.pop();
for(int i =0;i<4;i++){
int a = x + dx[i],b = y+dy[i];//移动
if(a>=0&&b>=0&&a<m&&b<n&&'#'!=grid[a][b]){//边界保护
char c = grid[a][b];//暂存字符
if(islower(c)){
int t = k;//暂存k
t |= 1<<(c-'a');//加入新的钥匙
if(t==(1<<cnt) - 1) return dis[x][y][k]+1;//第一转移到全部钥匙的状态//返回答案
if(-1 == dis[a][b][t]){//新状态没有搜索过。
dis[a][b][t] = dis[x][y][k] + 1;
// if(t==(1<<cnt)-1) return dis[a][b][t];//第一转移到全部钥匙的状态//返回答案
q.emplace(a,b,t);
}
}else{//起点,空房间 || isupper(grid[a][b])
if((('.'==c||'@'==c)||(k>>(c-'A'))&1)&&-1==dis[a][b][k]){锁对应的钥匙存在
dis[a][b][k] = dis[x][y][k] + 1;
q.emplace(a,b,k);
}
}
}
}
}
return -1;//无法获取所有钥匙
}
};
提交结果:
![](https://img-blog.csdnimg.cn/8d9f5f05e1d848049963f2264eb8f597.png)