【leetcode】【medium】752. 打开转盘锁​​​​​​​

752. 打开转盘锁

你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有10个数字： '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' 。每个拨轮可以自由旋转：例如把 '9' 变为  '0'，'0' 变为 '9' 。每次旋转都只能旋转一个拨轮的一位数字。

锁的初始数字为 '0000' ，一个代表四个拨轮的数字的字符串。

列表 deadends 包含了一组死亡数字，一旦拨轮的数字和列表里的任何一个元素相同，这个锁将会被永久锁定，无法再被旋转。

字符串 target 代表可以解锁的数字，你需要给出最小的旋转次数，如果无论如何不能解锁，返回 -1。

示例 1:

输入：deadends = ["0201","0101","0102","1212","2002"], target = "0202"
输出：6
解释：
可能的移动序列为 "0000" -> "1000" -> "1100" -> "1200" -> "1201" -> "1202" -> "0202"。
注意 "0000" -> "0001" -> "0002" -> "0102" -> "0202" 这样的序列是不能解锁的，
因为当拨动到 "0102" 时这个锁就会被锁定。

示例 2:

输入: deadends = ["8888"], target = "0009"
输出：1
解释：
把最后一位反向旋转一次即可 "0000" -> "0009"。

示例 3:

输入: deadends = ["8887","8889","8878","8898","8788","8988","7888","9888"], target = "8888"
输出：-1
解释：
无法旋转到目标数字且不被锁定。

示例 4:

输入: deadends = ["0000"], target = "8888"
输出：-1 

提示：

1. 死亡列表 deadends 的长度范围为 [1, 500]。
2. 目标数字 target 不会在 deadends 之中。
3. 每个 deadends 和 target 中的字符串的数字会在 10,000 个可能的情况 '0000' 到 '9999' 中产生。

题目链接：https://leetcode-cn.com/problems/open-the-lock/

 

法一：BFS

每个状态下有8个方向可走，并且用一个set记录已经走过的状态。

记忆：bfs的模版可通用。

class Solution {
public:
    string next(string cur, int idx, int gap){
        int num = cur[idx] - '0';
        if(cur[idx]=='9' && gap>0){
           cur[idx] = '0';
        }else if(cur[idx]=='0' && gap<0){
            cur[idx] = '9';
        }else{
            cur[idx] = cur[idx]+gap;
        }
        return cur;
    }
    int openLock(vector<string>& deadends, string target) {
        unordered_set<string> dead(deadends.begin(), deadends.end());
        int step=0;
        queue<string> q;
        q.push("0000");
        unordered_set<string> visit;
        visit.insert("0000");
        vector<int> gap={1,-1};
        while(!q.empty()){
            int len = q.size();
            for(int i=0; i<len; ++i){
                string cur = q.front();
                q.pop();
                if(cur==target){
                    return step;
                }
                if(dead.find(cur)!=dead.end()){
                    continue;
                }
                for(int i=0; i<4; ++i){
                    for(int j=0; j<2; ++j){
                        string s1 = next(cur, i, gap[j]);
                        if(visit.find(s1)==visit.end()){
                            q.push(s1);
                            visit.insert(s1);
                        }
                    }
                }
            }
            ++step;
        }
        return -1;
    }
};

法二：双向BFS

双向bfs能减少搜索过程中对无用节点的遍历，每次从节点较少一方的角度去搜索，停止条件是中途相遇。

因为要频繁查找，所以双向bfs不用queue，改用hash表。

class Solution {
public:
    string next(string cur, int idx, int gap){
        int num = cur[idx] - '0';
        if(cur[idx]=='9' && gap>0){
           cur[idx] = '0';
        }else if(cur[idx]=='0' && gap<0){
            cur[idx] = '9';
        }else{
            cur[idx] = cur[idx]+gap;
        }
        return cur;
    }
    int openLock(vector<string>& deadends, string target) {
        unordered_set<string> dead(deadends.begin(), deadends.end());
        int step=0;
        unordered_set<string> q1, q2, visit;
        q1.insert("0000");
        q2.insert(target);
        visit.insert("0000");
        vector<int> gap={1,-1};
        while(!q1.empty() && !q2.empty()){
            if(q1.size()>q2.size()){
                auto tmp = q2;
                q2 = q1;
                q1 = tmp;
            }
            unordered_set<string> q3;
            for(string cur: q1){
                if(q2.find(cur)!=q2.end()){
                    return step;
                }
                if(dead.find(cur)!=dead.end()){
                    continue;
                }
                for(int i=0; i<4; ++i){
                    for(int j=0; j<2; ++j){
                        string s = next(cur, i, gap[j]);
                        if(visit.find(s)==visit.end()){
                            q3.insert(s);
                        }
                    }
                }
            }
            q1 = q3;
            ++step;
        }
        return -1;
    }
};