Leetcode Index

864.Shortest Path to Get All Keys

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

给定一个二维网格 grid。 "." 代表一个空房间, "#" 代表一堵墙, "@" 是起点,("a", "b", ...)代表钥匙,("A", "B", ...)代表锁。 我们从起点开始出发,一次移动是指向四个基本方向之一行走一个单位空间。我们不能在网格外面行走,也无法穿过一堵墙。如果途经一个钥匙,我们就把它捡起来。除非我们手里有对应的钥匙,否则无法通过锁。 假设 K 为钥匙/锁的个数,且满足 1 <= K <= 6,字母表中的前 K 个字母在网格中都有自己对应的一个小写和一个大写字母。换言之,每个锁有唯一对应的钥匙,每个钥匙也有唯一对应的锁。另外,代表钥匙和锁的字母互为大小写并按字母顺序排列。 返回获取所有钥匙所需要的移动的最少次数。如果无法获取所有钥匙,返回 -1 。

方法:采用BFS方法,并通过将位置坐标和钥匙通过位运算保存到一个数里。 1. 首先获取输入向量的尺寸,并创建一个三维向量,因为6把钥匙共有2^6=64中状态,所以三维向量尺寸为m*n*64,表示到达每个位置的状。 2. 获取起始点,获取所有钥匙,设置起始点状态为1,表示已经访问过。 3. 当队列不为空时,去除队列中的元素,并拆分成横纵坐标以及钥匙状态,如此时要是状态和所有钥匙一致,则表示所有钥匙都访问到了,直接返回步数。 4. 四种情况,上下左右访问,如果越界或者碰壁则跳过。 5. 如果是锁,则判断是否已经有钥匙了,没有则跳过。 6. 如果是钥匙,则加入到当前的nkeys中,然后判断该位置nkeys是否访问过,如果访问过表示之前钥匙已经添加过了,跳过循环。 7. 将横纵坐标和钥匙状态压进队列,并将状态置1. 8. 等size--循环结束,将步长加1,继续判断队列是否为空,不为空继续循环。 9. 最后循环结束,没有符合keys和all_keys相等的情况,返回-1。

class Solution {
public:
int shortestPathAllKeys(vector<string>& grid) {
int m=grid.size(), n=grid[0].size();
queue<int> q;
vector<vector<vector<int>>> state(m,vector<vector<int>>(n,vector<int>(64,0)));
int all_keys=0;
int go[]={-1,0,1,0,-1};
for(int i=0;i<m;i++)
for(int j=0;j<n;j++){
auto c=grid[i][j];
if(c=='@')
{
q.push(i<<16 | j<<8);
state[i][j][0]=1;
}
if(c>='a' && c<='f')
all_keys |= 1 << (c-'a');
}
int step=0;
while(!q.empty())
{
int size=q.size();
while(size--)
{
int s=q.front();
q.pop();
int x=s>>16;
int y = s>>8 & 0xFF;
int keys=s & 0xFF;
if(keys == all_keys)
return step;
for(int i=0;i<4;i++)
{
int nx= x+ go[i];
int ny= y+ go[i+1];
if(nx <0 || nx >=m ||ny <0 || ny >=n) continue;
auto c = grid[nx][ny];
int nkeys=keys;
if(c=='#') continue;
if(c>='A' && c<='F' && !((1<<(c-'A')) & keys) ) continue;
if(c>='a' && c<='f') nkeys |= 1<<(c-'a');
if(state[nx][ny][nkeys]) continue;
q.push((nx <<16) | (ny <<8) | nkeys);
state[nx][ny][nkeys]=1;
}
}
step++;
}
return -1;
}
};