752. 打开转盘锁

链接:

752. 打开转盘锁

题解:

cpp 复制代码
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()
                || table.find(target) != table.end()) {
            return -1;
        }
        std::queue<std::string> que;
        std::unordered_map<std::string, int> distance;
        que.push(std::string("0000"));
        distance["0000"] = 0;
        auto get_next = [](std::string& word)->std::vector<std::string> {
            std::vector<std::string> result;
            for (int i = 0; i < word.size(); ++i) {
                char ch = word[i];
                if (word[i] == '9') {
                    word[i] = '0';
                } else {
                    ++word[i];
                }
                result.push_back(word);
                word[i] = ch;
                if (word[i] == '0') {
                    word[i] = '9';
                } else {
                    --word[i];
                }
                result.push_back(word);
                word[i] = ch;
            }
            return result;
        };
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; ++i) {
                auto f = que.front();
                que.pop();
                if (f == target) {
                    return distance[f];
                }
                for (auto& next : get_next(f)) {
                    //cout << "next = " << next << endl;
                    if (table.find(next) != table.end()) {
                        continue;
                    }
                    if (distance.find(next) != distance.end()) {
                        continue;
                    }
                    /*if (next == target) {
                        return distance[f] + 1;
                    }*/
                    distance[next] = distance[f] + 1;
                    que.push(next);
                }
            }
        }
        return -1;
    }
};
cpp 复制代码
class Solution {
public:
    int openLock(vector<string>& deadends, string target) {
        if (target == "0000") {
            return 0;
        }

        unordered_set<string> dead(deadends.begin(), deadends.end());
        if (dead.count("0000")) {
            return -1;
        }

        auto num_prev = [](char x) -> char {
            return (x == '0' ? '9' : x - 1);
        };
        auto num_succ = [](char x) -> char {
            return (x == '9' ? '0' : x + 1);
        };

        // 枚举 status 通过一次旋转得到的数字
        auto get = [&](string& status) -> vector<string> {
            vector<string> ret;
            string bak = status;
            for (int i = 0; i < 4; ++i) {
                char num = status[i];
                status[i] = num_prev(num);
                ret.push_back(status);
                status[i] = num_succ(num);
                ret.push_back(status);
                status = bak;
            }
            return ret;
        };

        queue<pair<string, int>> q;
        q.emplace("0000", 0);
        unordered_set<string> seen = {"0000"};
        int step = 0;
        while (!q.empty()) {
            auto [status, step] = q.front();
            q.pop();
            for (auto&& next_status: get(status)) {
                if (!seen.count(next_status) && !dead.count(next_status)) {
                    if (next_status == target) {
                        //return step + 1;
                        return step+1;
                    }
                    q.emplace(next_status, step + 1);
                    seen.insert(move(next_status));
                }
               
            }
            ++step;
        }

        return -1;
    }
};
相关推荐
Fanxt_Ja3 小时前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下3 小时前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶4 小时前
算法 --- 字符串
算法
博笙困了4 小时前
AcWing学习——差分
c++·算法
NAGNIP4 小时前
认识 Unsloth 框架:大模型高效微调的利器
算法
NAGNIP4 小时前
大模型微调框架之LLaMA Factory
算法
echoarts4 小时前
Rayon Rust中的数据并行库入门教程
开发语言·其他·算法·rust
Python技术极客4 小时前
一款超好用的 Python 交互式可视化工具,强烈推荐~
算法
徐小夕4 小时前
花了一天时间,开源了一套精美且支持复杂操作的表格编辑器tablejs
前端·算法·github
小刘鸭地下城4 小时前
深入浅出链表:从基础概念到核心操作全面解析
算法