LeetCode 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' 中产生。

---

重新写了一遍这道题，上一次代码

对BFS理解的稍微好一点了，每个点都是状态，保存的时候直接保存一个点的状态就可以了，以前的set现在换成了map，这个没什么区别，但是时间慢了很多，可能是服务器的原因，每次和每次都不一样

class Solution
{
  private:
    struct point
    {
        int step;
        string num;
        point(string num = "", int step = 0) : num(num), step(step) {}
    };
    unordered_map<string, bool> dead, vis;

  public:
    int openLock(vector<string> &deadends, string target)
    {
        for (string str : deadends)
            dead[str] = true;
        if (dead["0000"])
            return -1;

        vis["0000"] = true;
        queue<point> q;
        q.push(point{"0000", 0});

        while (!q.empty())
        {
            point cur = q.front();
            q.pop();
            if (cur.num == target)
                return cur.step;
            for (int i = 0; i < 4; ++i)// 对每个位置都操作一遍
            {
                string str = cur.num;
                str[i] = (str[i] - '0' + 1) % 10 + '0';
                if (!dead[str] && !vis[str])
                {
                    vis[str] = true;
                    q.push(point{str, cur.step + 1});
                }
                str = cur.num;
                str[i] = (str[i] - '0' + 9) % 10 + '0';//减法换成加法
                if (!dead[str] && !vis[str])
                {
                    vis[str] = true;
                    q.push(point{str, cur.step + 1});
                }
            }
        }
        return -1;
    }
};

不过用回set确实快，内存也少，也要注意找不到是== set.end()

class Solution
{
  private:
    struct point
    {
        int step;
        string num;
        point(string num = "", int step = 0) : num(num), step(step) {}
    };
    unordered_set<string> dead, vis;

  public:
    int openLock(vector<string> &deadends, string target)
    {
        for (string str : deadends)
            dead.insert(str);
        if (dead.find("0000") != dead.end())//能找到
        {
            cout << 1;
            return -1;
        }

        vis.insert("0000");
        queue<point> q;
        q.push(point{"0000", 0});

        while (!q.empty())
        {
            point cur = q.front();
            q.pop();
            if (cur.num == target)
                return cur.step;
            for (int i = 0; i < 4; ++i)
            {
                string str = cur.num;
                str[i] = (str[i] - '0' + 1) % 10 + '0';
                if (dead.find(str) == dead.end() && vis.find(str) == vis.end())//找不到
                {
                    vis.insert(str);
                    q.push(point{str, cur.step + 1});
                }
                str = cur.num;
                str[i] = (str[i] - '0' + 9) % 10 + '0';
                if (dead.find(str) == dead.end() && vis.find(str) == vis.end())
                {
                    vis.insert(str);
                    q.push(point{str, cur.step + 1});
                }
            }
        }
        return -1;
    }
};