题目:
一个密码锁由 4 个环形拨轮组成,每个拨轮都有 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
分析:
如果一个问题是关于某事务状态的改变,那么可以考虑把问题转换成图搜索问题,事务的每个状态是图的一个节点,如果一个状态能够转变为另一个状态,那么这两个状态对应的节点之间有一条边相连。
对于该问题而言,密码锁的每个状态都对应着图中的一个节点,如状态“0000”是一个节点,“0001”是另一个节点。如果转动某个转轮一次可以让密码锁从一个状态转移到另一个状态,那么这两个状态之间有一条边相连。例如将状态“0000”分别向上或向下转动4个转轮中的一个,可以得到8个状态,即0001,0009,0010,0090,0100,0900,1000,9000,那么图中的节点“0000”就有8条边分别和这8个状态对应的节点相连。
由于题目中要求找出节点“0000”到密码的对应节点的最短路径的长度,因此应该采用广度优先搜索。这是因为广度优先搜素是从起始节点开始首先达到所有距离为1的节点,接着到达所有距离为2的节点。广度优先搜索一定是从起始节点沿着最短路径到达目标节点的。
具体细节见代码注释
代码:
package com.kuang;
import java.util.*;
public class OpenLock {
public int openLock(String[] deadends, String target) {
Set<String> dead = new HashSet<>(Arrays.asList(deadends));
Set<String> visited = new HashSet<>();
String init = "0000";
// 如果死亡数字中包含init或target直接返回-1
if (dead.contains(init)||dead.contains(target)){
return -1;
}
// 队列queue1中存放的是需要转动n次到达的节点
Queue<String> queue1 = new LinkedList<>();
// 队列queue2中存放的是和queue1中的节点相连但还没有搜索到的节点
Queue<String> queue2 = new LinkedList<>();
int steps = 0;
queue1.offer(init);
visited.add(init);
while (!queue1.isEmpty()){
// 出队节点
String cur = queue1.remove();
if (cur.equals(target)){
return steps;
}
// 获取该节点可以转化的所有状态
List<String> neighbors = getNeighbors(cur);
for (String neighbor : neighbors) {
// 去掉死亡状态和已经访问状态
if (!dead.contains(neighbor)&&!visited.contains(neighbor)){
// 满足条件的状态节点入队列2
queue2.add(neighbor);
visited.add(neighbor);
}
}
if (queue1.isEmpty()){
steps++;
queue1 = queue2;
queue2 = new LinkedList<>();
}
}
return -1;
}
private List<String> getNeighbors(String cur) {
List<String> nexts = new LinkedList<>();
for (int i = 0; i < cur.length(); i++) {
char ch = cur.charAt(i);
char newCh = ch == '0'?'9':(char)(ch-1);
StringBuilder stringBuilder = new StringBuilder(cur);
stringBuilder.setCharAt(i,newCh);
nexts.add(stringBuilder.toString());
newCh = ch == '9'?'0':(char)(ch+1);
stringBuilder.setCharAt(i,newCh);
nexts.add(stringBuilder.toString());
}
return nexts;
}
}