【重点!!!】【BFS】752.打开转盘锁

Allenlzcoder

于 2024-01-02 15:28:50 发布

阅读量309

点赞数 6

分类专栏： labuladong的算法小抄笔记文章标签： BFS 打开转盘锁

本文链接：https://blog.csdn.net/Allenlzcoder/article/details/135342056

版权

labuladong的算法小抄笔记专栏收录该内容

77 篇文章 0 订阅

订阅专栏

题目

法1：BFS

基础算法，必须掌握!!!

class Solution {
    public int openLock(String[] deadends, String target) {
        if ("0000".equals(target)) {
            return 0;
        }
        Set<String> deadSet = new HashSet<>();
        Set<String> usedSet = new HashSet<>();
        for (String tmp : deadends) {
            if ("0000".equals(tmp)) {
                return -1;
            }
            deadSet.add(tmp);
        }
        int step = 0;
        Queue<String> queue = new LinkedList<>();
        queue.offer("0000");
        usedSet.add("0000");
        while (!queue.isEmpty()) {
            ++step;
            int size = queue.size();
            for (int i = 0; i < size; ++i) { // 类似二叉树层次遍历，每层节点数量
                String curStatus = queue.poll();
                for (String nextTmp : nextStatus(curStatus)) {
                    if (!deadSet.contains(nextTmp) && !usedSet.contains(nextTmp)) {
                        if (nextTmp.equals(target)) {
                            return step;
                        }
                        queue.offer(nextTmp);
                        usedSet.add(nextTmp);
                    }
                }
            }
        }

        return -1; //无法实现返回-1
    }

    public List<String> nextStatus(String cur) {
        List<String> res = new ArrayList<>();
        char[] array = cur.toCharArray();
        for (int i = 0; i < 4; ++i) {
            char tmp = array[i];
            char plusChar = plusOne(tmp);
            char minusChar = minusOne(tmp);
            array[i] = plusChar;
            res.add(String.valueOf(array));
            array[i] = minusChar;
            res.add(String.valueOf(array));
            array[i] = tmp; // 切记这里一定要复原!!!
        }

        return res;
    }

    public char plusOne(char c) {
        return c == '9' ? '0' : (char) (c + 1);
    }

    public char minusOne(char c) {
        return c == '0' ? '9' : (char) (c - 1);
    }
}