你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有10个数字: ‘0’, ‘1’, ‘2’, ‘3’, ‘4’, ‘5’, ‘6’, ‘7’, ‘8’, ‘9’ 。每个拨轮可以自由旋转:例如把 ‘9’ 变为 ‘0’,‘0’ 变为 ‘9’ 。每次旋转都只能旋转一个拨轮的一位数字。
锁的初始数字为 ‘0000’ ,一个代表四个拨轮的数字的字符串。
列表 deadends 包含了一组死亡数字,一旦拨轮的数字和列表里的任何一个元素相同,这个锁将会被永久锁定,无法再被旋转。
字符串 target 代表可以解锁的数字,你需要给出解锁需要的最小旋转次数,如果无论如何不能解锁,返回 -1 。
题解
方法一:广度优先搜索
思路与算法
我们可以使用广度优先搜索,找出从初始数字 0000 到解锁数字 target 的最小旋转次数。
具体地,我们在一开始将 (0000,0) 加入队列,并使用该队列进行广度优先搜索。在搜索的过程中,设当前搜索到的数字为 status,旋转的次数为 step,我们可以枚举 status 通过一次旋转得到的数字。设其中的某个数字为 next_status,如果其没有被搜索过,我们就将 (next_status,step+1) 加入队列。如果搜索到了 target,我们就返回其对应的旋转次数。
为了避免搜索到死亡数字,我们可以使用哈希表存储 deadends 中的所有元素,这样在搜索的过程中,我们可以均摊 O(1) 地判断一个数字是否为死亡数字。同时,我们还需要一个哈希表存储所有搜索到的状态,避免重复搜索。
如果搜索完成后,我们仍没有搜索到 target,说明我们无法解锁,返回 −1。
细节
本题中需要注意如下两个细节:
如果 target 就是初始数字 000,那么直接返回答案 0;
如果初始数字 0000 出现在 deadends 中,那么直接返回答案 −1。
class Solution {
public int openLock(String[] deadends, String target) {
//如果 target 就是初始数字 0000,那么直接返回答案 0
if ("0000".equals(target)) {
return 0;
}
//使用哈希表存储 deadends 中的所有元素,可以O(1)地判断一个数字是否为死亡数字
Set<String> dead = new HashSet<String>();
for (String deadend : deadends) {
dead.add(deadend);
}
//如果初始数字 0000 出现在 deadends 中,那么直接返回答案 −1
if (dead.contains("0000")) {
return -1;
}
//旋转的次数为 step
int step = 0;
Queue<String> queue = new LinkedList<String>();
queue.offer("0000");
//哈希表存储所有搜索到的状态,避免重复搜索
Set<String> seen = new HashSet<String>();
seen.add("0000");
while (!queue.isEmpty()) {
++step;
int size = queue.size();
for (int i = 0; i < size; ++i) {
//设当前搜索到的数字为 status
String status = queue.poll();
//需要 O(d) 的时间生成旋转后的数字(即加入队列)
for (String nextStatus : get(status)) {
if (!seen.contains(nextStatus) && !dead.contains(nextStatus)) {
if (nextStatus.equals(target)) {
return step;
}
queue.offer(nextStatus);
seen.add(nextStatus);
}
}
}
}
return -1;
}
public char numPrev(char x) {
return x == '0' ? '9' : (char) (x - 1);
}
public char numSucc(char x) {
return x == '9' ? '0' : (char) (x + 1);
}
// 枚举 status 通过一次旋转得到的数字
public List<String> get(String status) {
List<String> ret = new ArrayList<String>();
char[] array = status.toCharArray();
//status中的四位数字都要前后旋转
//需要 O(d) 的时间枚举旋转的数位,d是转盘数字的位数
for (int i = 0; i < 4; ++i) {
char num = array[i];
array[i] = numPrev(num);
ret.add(new String(array));
array[i] = numSucc(num);
ret.add(new String(array));
array[i] = num;
}
return ret;
}
}