1. 程式人生 > 實用技巧 >開啟轉盤鎖

開啟轉盤鎖

開啟轉盤鎖

題目:你有一個帶有四個圓形撥輪的轉盤鎖。每個撥輪都有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

解題思路:用BFS解決,當發現窮舉到的是deadends裡面的數字時不要讓其入隊

class Solution {
    public int openLock(String[] deadends, String target) {
        Set<String> deads = new HashSet(Arrays.asList(deadends));
        if(deads.contains("0000"))
            return -1;      
        Set<String> visited = new HashSet();
        Queue<String> queue = new LinkedList();
        queue.offer("0000");
        visited.add("0000");
        int ans = 0;
        while(!queue.isEmpty()) {
            int size = queue.size();
            for(int j = 0; j < size; j++) {
                String cur = queue.poll();               
                if(cur.equals(target)) {
                    return ans;
                }                
                for(int i = 0; i < 4; i++) {
                    String upOne = addOne(cur, i);
                    String downOne = subOne(cur, i);
                    if(!visited.contains(upOne) && !deads.contains(upOne)) {
                        visited.add(upOne);
                        queue.offer(upOne);
                    }
                    // System.out.println("up = " + upOne);
                    // System.out.println("down = " + downOne);
                    if(!visited.contains(downOne) && !deads.contains(downOne)) {
                        visited.add(downOne);
                        queue.offer(downOne);
                    }
                }
            }
            
            ans++;
        }
        
        return -1;
    }
    
    private String addOne(String s, int j) {
        char[] ch = s.toCharArray();
        if (ch[j] == '9')
            ch[j] = '0';
        else
            ch[j] += 1;
        return new String(ch);
    }
    
    private String subOne(String s, int j) {
        char[] ch = s.toCharArray();
        if (ch[j] == '0')
            ch[j] = '9';
        else
            ch[j] -= 1;
        return new String(ch);
    }
}