剑指 Offer II 109. 开密码锁

剑指 Offer II 109. 开密码锁

一个密码锁由 4 个环形拨轮组成，每个拨轮都有 10 个数字： ‘0’, ‘1’, ‘2’, ‘3’, ‘4’, ‘5’, ‘6’, ‘7’, ‘8’, ‘9’ 。每个拨轮可以自由旋转：例如把 ‘9’ 变为 ‘0’，‘0’ 变为 ‘9’ 。每次旋转都只能旋转一个拨轮的一位数字。

锁的初始数字为 ‘0000’ ，一个代表四个拨轮的数字的字符串。

列表 deadends 包含了一组死亡数字，一旦拨轮的数字和列表里的任何一个元素相同，这个锁将会被永久锁定，无法再被旋转。

字符串 target 代表可以解锁的数字，请给出解锁需要的最小旋转次数，如果无论如何不能解锁，返回 -1 。

输入：deadends = ["0201","0101","0102","1212","2002"], target = "0202"
输出：6
解释：
可能的移动序列为 "0000" -> "1000" -> "1100" -> "1200" -> "1201" -> "1202" -> "0202"。
注意 "0000" -> "0001" -> "0002" -> "0102" -> "0202" 这样的序列是不能解锁的，因为当拨动到 "0102" 时这个锁就会被锁定。

思路1：从0000出发，每次都有四种选择，即第i层的节点信息是8^(i-1)个，把整个情况想成一棵树，这颗树的每个节点都有8个子节点，我们利用BFS进行一层一层的遍历，如果遇到dead密码，我们就跳过这个节点以及以这个节点为根的子树。但是现在还有一个问题，就是我们保证节点不重复，例如0000的节点有0001，但是0001的节点又有0000，导致这棵树无穷无尽了。所以我们引入一个visit数组用来记录已经访问过的密码，避免走回头路。

class Solution {
    public int openLock(String[] deadends, String target) {
        // 分别用来记录无法访问 以及 已经访问过的密码
        Set<String> dead = new HashSet<>();
        Set<String> visit = new HashSet<>();
        // 队列 
        Queue<String> q = new LinkedList<>();
        // 步长
        int step = 0;
        for(String s:deadends) dead.add(s);
        q.offer("0000");
        visit.add("0000");
        while(!q.isEmpty()){
            // 获取每一层的节点个数
            int length = q.size();
            // 将这一层的所有满足的字节点加入到队列中
            for(int i = 0;i < length;i++){
                String temp = q.poll();
                // 如果是dead密码 就跳过这一层的这个节点下面的分支，如果是正确的密码 就返回
                if(dead.contains(temp)) continue;
                if(temp.equals(target)) return step;
                for(int j = 0;j < 4;j++){
                    String down = down_password(temp,j);
                    String up = up_password(temp,j);
                    // 未访问过的才能加入
                    if(!visit.contains(down)) {
                        q.offer(down);
                        visit.add(down);    
                    }
                    if(!visit.contains(up)) {
                        q.offer(up);
                        visit.add(up);    
                    }         
                }
            }
            step++;
        }
        return -1;
    }

    String up_password(String s,int index){
        char[] ch = s.toCharArray();
        if(ch[index] == '9') ch[index] = '0';
        else    ch[index] += 1;
        return new String(ch);
    }

    String down_password(String s,int index){
        char[] ch = s.toCharArray();
        if(ch[index] == '0') ch[index] = '9';
        else    ch[index] -= 1;
        return new String(ch);
    }
}