剑指 Offer II 109. 开密码锁

INGNIGHT

已于 2022-07-25 19:58:48 修改

阅读量251

点赞数 1

分类专栏： leetcode 文章标签： c++ 算法开发语言

于 2022-07-25 19:58:46 首次发布

本文链接：https://blog.csdn.net/INGNIGHT/article/details/125982214

版权

leetcode 专栏收录该内容

508 篇文章 1 订阅

订阅专栏

这是一个关于解决开密码锁问题的算法实现。代码中使用了广度优先搜索（BFS）策略，通过一个队列来存储当前状态，并利用unordered_set避免重复访问和检查死路。算法首先排除已知的死路，然后从初始状态开始，逐层遍历所有可能的密码组合，直到找到目标密码。get_neighboards函数用于生成当前密码的所有相邻状态。如果找到目标密码，返回步数，否则返回-1表示无法到达。

摘要由CSDN通过智能技术生成

链接：剑指 Offer II 109. 开密码锁

题解：

class Solution {
public:
    int openLock(vector<string>& deadends, string target) {
        std::unordered_set<std::string> table(deadends.begin(), deadends.end());
        // 面向异常设计
        if (table.find("0000") != table.end()) {
            return -1;
        }
        // 面向异常设计
        if (target == "0000") {
            return 0;
        }
        std::queue<std::pair<std::string, int>> que;
        que.push(make_pair("0000", 0));
        std::unordered_set<std::string> visited;
        visited.insert("0000");
        while (!que.empty()) {
            auto f = que.front();
            que.pop();
            for (auto& neighboard : get_neighboards(f.first)) {
                if (visited.find(neighboard) != visited.end() || table.find(neighboard) != table.end()) {
                    continue;
                }
                if (neighboard == target) {
                    return f.second+1;
                }
                que.push(std::make_pair(neighboard, f.second+1));
                visited.insert(neighboard);
            }
        }
        return -1;
    }
private:
    std::vector<std::string> get_neighboards(std::string& cur) {
        auto next = [](char a)->char {
            return a == '9' ? '0' : a+1;
        };
        auto prev = [](char a)->char {
            return a == '0' ? '9' : a-1;
        };
        std::vector<std::string> result;
        for (int i = 0; i < cur.size(); ++i) {
            char c = cur[i];
            cur[i] = next(c);
            result.push_back(cur);
            cur[i] = prev(c);
            result.push_back(cur);
            cur[i] = c;
        }
        return result;
    }
};