leetcode0752. 打开转盘锁-medium

1 题目：打开转盘锁

官方标定难度：中

你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有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
解释：无法旋转到目标数字且不被锁定。

提示：

1 <= deadends.length <= 500
deadends[i].length == 4
target.length == 4
target 不在 deadends 之中
target 和 deadends[i] 仅由若干位数字组成

2 solution

采用广度优先搜索，直到搜到 target 或者没有下一种情况可以搜索到

代码

class Solution {
    /*
     * 从 0000 开始 -> target
     */
public:
    int openLock(vector<string> &deadends, string target) {
        queue<string> q[2];
        unordered_set<string> set;
        for (auto s: deadends) set.insert(s);
    
        if(set.count("0000")) return -1;
        if(target == "0000") return 0;

        q[0].push("0000");
        set.insert("0000");
        

        for (int i = 0;; i++) {
            while (!q[i & 1].empty()) {
                string x = q[i & 1].front();
                //cout << x << " ";
                q[i & 1].pop();
                for (int k = 0; k < 4; k++) {
                    string xx = x;
                    xx[k] = (xx[k] - '0' + 1) % 10 + '0';
                    if(xx == target) return i + 1;
                    if (!set.count(xx)){
                        set.insert(xx);
                        q[(i + 1) & 1].push(xx);
                    }
                    xx[k] = (xx[k] - '0' + 8) % 10 + '0';
                    if(xx == target) return i + 1;
                    if (!set.count(xx)){
                        set.insert(xx);
                        q[(i + 1) & 1].push(xx);
                    }
                }
            }
            if(q[(i + 1) & 1].empty()) return -1;
        }
    }
};

结果

3 进阶

用整数表示以提高计算效率，用 bitset 提高空间利用率

代码

class Solution {
    /*
     * 换成整数优化
     */
public:
    int openLock(vector<string> &deadends, string target) {
        queue<int> q[2];
        bitset<10000> set;

        int tar = 0;
        for(char c : target) tar = tar * 10 + c - '0';
        
        for (auto s: deadends) {
            int num = 0;
            for(char c : s){
                num = num * 10 + c - '0';
            }
            set.set(num);
        }
        
        if(set.test(0)) return -1;
        if(tar == 0) return 0;

        q[0].push(0);
        set.set(0);


        for (int i = 0;; i++) {
            while (!q[i & 1].empty()) {
                int x = q[i & 1].front();
                q[i & 1].pop();
                int p = 1;
                for (int k = 0; k < 4; k++) {
                    int digit = x % (p * 10) / p;
                    int xx = x - digit * p;
                    digit = (digit + 1) % 10;
                    
                    if(xx + digit * p == tar) return i + 1;
                    if (!set.test(xx + digit * p)){
                        set.set(xx + digit * p);
                        q[(i + 1) & 1].push(xx + digit * p);
                    }
                    
                    digit = (digit + 8) % 10;
                    if(xx + digit * p == tar) return i + 1;
                    if (!set.test(xx + digit * p)){
                        set.set(xx + digit * p);
                        q[(i + 1) & 1].push(xx + digit * p);
                    }
                    p *= 10;
                }
            }
            if(q[(i + 1) & 1].empty()) return -1;
        }
    }
};

结果