864.获取所有钥匙的最短路径(BFS+位图技巧)

给定一个二维网格 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] 
  • 每个钥匙都对应一个 不同 的字母
  • 每个钥匙正好打开一个对应的锁

思路:BFS+状态压缩(位图)

首先,遇到二维数组的BFS,应该用queue队列存放每个点的坐标(x, y)作目前状态,用二维数组dire={{-1,0},{1,0},{0,-1},{0,1}}作上下左右四个方向的选择。此外,还需要使用二维数组visited记录已经遍历过的点,防止走回头路,未遍历的记作visited[x][y]=false,遍历过的记作visited[x][y]=true,类似题目见934. 最短的桥 - 力扣(LeetCode)

 但是这道题中,是可以走回头路的,这取决于你拿到钥匙的状态。如果例一中,第一行的钥匙在最后一格,那么为了拿到所有的钥匙,就要先向右走到头,再向左走从两堵墙中间下去,这就走了回头路。

所以visited数组应该是一个三维数组,包括了该点的坐标以及此时拿到钥匙的状态,意为该点(x, y) 是可以重复经过的,但是重复经过时拿到钥匙的状态应该有所不同。

visited(m,vector<vector<bool>>(n,vector<bool>(allkey+1,false)))

同理,压入队列中的每一种状态也应该是该点的坐标(x, y)+拿到钥匙的状态

struct State {
    int x;
    int y;
    int key;   
};

我们使用int key 来表示拿到钥匙的状态,key的二进制表示为1010时,表示拿到第2把和第4把钥匙。下列是一些有用的位图技巧:

1、将第m位置1   (m>=1)                          key | = 1<< (m-1)                                                            

2、判断第m位是否是1    (m>=1)              ( key >> (m-1) ) & 1) ==0 时不为1 ,==1 时为1

struct State {
    int x;
    int y;
    int key;
};
class Solution {
public:
    int shortestPathAllKeys(vector<string>& grid) {
        State start;
        int keycnt=0;
        int m=grid.size();
        int n=grid[0].size();
        queue<State> q;
        vector<vector<int>> dire={{-1,0},{1,0},{0,-1},{0,1}};
        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(grid[i][j]>='a'&&grid[i][j]<='z')//记录钥匙数
                {
                    keycnt++;
                }
            }
        }
        q.push(start);//初始状态入队
        int allkey=(1<<keycnt)-1;//表示所有钥匙都拿到的状态
        //keycnt=2,1<<keycnt表示1向左移动2位,是二进制的100,100-001等于二进制表示的011,表示第一位和第二位有钥匙
        vector<vector<vector<bool>>> visited(m,vector<vector<bool>>(n,vector<bool>(allkey+1,false)));
        //visited数组,记录遍历过程中的状态,防止走回头路。(“状态”是指在点(x,y)处拥有钥匙的状态)
        int step=0;
        while(!q.empty())
        {
            int size=q.size();
            for(int i=0;i<size;i++)
            {
                State cur=q.front();
                q.pop();
                if(cur.key==allkey)
                    return step;
                for(int i=0;i<dire.size();i++)
                {
                    int x=cur.x+dire[i][0];
                    int y=cur.y+dire[i][1];
                    int key=cur.key;
                    if(x<0||y<0||x>=m||y>=n)//越界
                        continue;

                    char c=grid[x][y];
                    if(c=='#')//墙
                        continue;

                    if(c>='A'&&c<='Z')//遇到锁
                    {
                        //遇到锁'B',判断有无钥匙'b',即判断key的第二位是否为1
                        //将要判断的那位移到末尾,再与001相与,为000,则该位是0,为001,则该位是1
                        if((key>>(c-'A')&1)==0)//若key=1001,则1001>>('B'-'A')=0100,0100&0001=0000,没有'b'钥匙
                        {
                            continue;
                        }
                    }

                    if(c>='a'&&c<='z')//捡到钥匙
                    {
                        //更新钥匙状态
                        key |=1<<(c-'a');//若得到钥匙'b',则1<<('b'-'a')是二进制的10,key与10相或,就将第二位置1
                    }

                    if(visited[x][y][key]==true)//已经遍历过
                        continue;

                    visited[x][y][key]=true;//更新已遍历过的情况
                    //将情况加入队列中
                    q.push(State{x,y,key});//C++11新语法,数据类型后面直接加{},给成员变量赋初值
                }
            }
            step++;
        }
        return -1;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值