LeetCode0752-打开转盘锁
题目:
你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有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
提示:
死亡列表 deadends 的长度范围为 [1, 500]。
目标数字 target 不会在 deadends 之中。
每个 deadends 和 target 中的字符串的数字会在 10,000 个可能的情况 '0000' 到 '9999' 中产生。
分析:
Solution01使用BFS算法,Solution02 使用双向BFS,双向BFS局限性是必须知道终点在哪,本题刚好合适。
/**
* 0752-打开转盘锁
* 你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有10个数字: '0', '1', '2', '3',
* '4', '5', '6', '7', '8', '9' 。每个拨轮可以自由旋转:例如把 '9' 变为 '0'
* ,'0' 变为 '9' 。每次旋转都只能旋转一个拨轮的一位数字。
* <p>
* 锁的初始数字为 '0000' ,一个代表四个拨轮的数字的字符串。
* <p>
* 列表 deadends 包含了一组死亡数字,一旦拨轮的数字和列表里的任何一个元素相同,
* 这个锁将会被永久锁定,无法再被旋转。
* <p>
* 字符串 target 代表可以解锁的数字,你需要给出最小的旋转次数,如果无论如何不能解
* 锁,返回 -1。
* <p>
* <p>
* <p>
* 示例 1:
* <p>
* 输入:deadends = ["0201","0101","0102","1212","2002"], target = "0202"
* 输出:6
* 解释:
* 可能的移动序列为 "0000" -> "1000" -> "1100" -> "1200" -> "1201" -> "1202" ->
* "0202"。
* 注意 "0000" -> "0001" -> "0002" -> "0102" -> "0202" 这样的序列是不能解锁的,
* 因为当拨动到 "0102" 时这个锁就会被锁定。
* <p>
* 示例 2:
* <p>
* 输入: deadends = ["8888"], target = "0009"
* 输出:1
* 解释:
* 把最后一位反向旋转一次即可 "0000" -> "0009"。
* <p>
* 示例 3:
* <p>
* 输入: deadends = ["8887","8889","8878","8898","8788","8988","7888","9888"],
* target = "8888"
* 输出:-1
* 解释:
* 无法旋转到目标数字且不被锁定。
* <p>
* 示例 4:
* <p>
* 输入: deadends = ["0000"], target = "8888"
* 输出:-1
* <p>
* 提示:
* <p>
* 死亡列表 deadends 的长度范围为 [1, 500]。
* 目标数字 target 不会在 deadends 之中。
* 每个 deadends 和 target 中的字符串的数字会在 10,000 个可能的情况 '0000' 到 '9999' 中产生。
*/
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Set;
/**
* BFS
*/
class Solution01 {
/**
* 解锁函数
*
* @param deadends 死亡数组
* @param target 目标值
* @return 需要旋转的次数
*/
public int openLock(String[] deadends, String target) {
// 记录需要跳过的死亡密码 -- 使用Set 是为了避免添加重复值
Set<String> deads = new HashSet<>();
for (String s : deadends) {
deads.add(s);
}
// 记录已经穷举过的密码,防止走回头路
Set<String> visited = new HashSet<>();
// 初始化队列
Queue<String> q = new LinkedList<>();
// 从起点开始启动广度优先搜索
// 保存需要的次数
int step = 0;
/*
*从锁的初始值开始
* offer() 如果在可以不违反容量限制的情况下立即将指定的元素插入此队列,
* 当使用容量受限的队列时,此方法通常比 add() 更可取,add()可能会通过引发
* 异常而无法插入元素。
*/
q.offer("0000");
visited.add("0000");
while (!q.isEmpty()) {
// 获取队列的长度
int sz = q.size();
// 将当前队列中的所有节点向周围扩散
for (int i = 0; i < sz; i++) {
// 获取对头元素,并出队列
String cur = q.poll();
// 判断是否到达终点
if (deads.contains(cur)) {
// 当前值在死亡数组中,则直接跳过
continue;
}
if (cur.equals(target)) {
// 当前值为目标值,则直接返回 step
return step;
}
// 将一个节点的未遍历相邻节点加入队列
for (int j = 0; j < 4; j++) {
// 向上旋转一次
String up = plusOne(cur, j);
// 判断,如果向上旋转一次,不在已经访问过的集合中,则将其加入队列,并加入访问过集合
if (!visited.contains(up)) {
q.offer(up);
visited.add(up);
}
// 向下旋转一次
String down = minusOne(cur, j);
if (!visited.contains(down)) {
q.offer(down);
visited.add(down);
}
}
}
// 这里增加步数
step++;
}
// 如果穷举完都没有找到目标代码,则找不到,返回-1
return -1;
}
/**
* 将cur[j]向上拨动一次
*
* @param cur 当前字符串
* @param j 需要向上拨动一次的位置
* @return 旋转后的字符串
*/
private String plusOne(String cur, int j) {
char[] ch = cur.toCharArray();
if (ch[j] == '9') {
ch[j] = '0';
} else {
ch[j] += 1;
}
return new String(ch);
}
/**
* 将cur[j]向下拨动一次
*
* @param cur 当前字符串
* @param j 需要向下拨动一次的位置
* @return 旋转后的字符串
*/
private String minusOne(String cur, int j) {
char[] ch = cur.toCharArray();
if (ch[j] == '0') {
ch[j] = '9';
} else {
ch[j] -= 1;
}
return new String(ch);
}
}
/**
* 双向BFS
* 局限--必须知道终点在哪,例如二叉树的最小高度问题则不能使用
*/
class Solution02{
/**
* 解锁函数
*
* @param deadends 死亡数组
* @param target 目标值
* @return 需要旋转的次数
*/
int openLock(String[] deadends, String target) {
Set<String> deads = new HashSet<>();
for (String s : deadends) deads.add(s);
// 用集合不用队列,可以快速判断元素是否存在
// q1从头开始
Set<String> q1 = new HashSet<>();
// q2 从目标开始
Set<String> q2 = new HashSet<>();
Set<String> visited = new HashSet<>();
int step = 0;
q1.add("0000");
q2.add(target);
while (!q1.isEmpty() && !q2.isEmpty()) {
// 哈希集合在遍历的过程中不能修改,用 temp 存储扩散结果
Set<String> temp = new HashSet<>();
// 将 q1 中的所有节点向周围扩散
for (String cur : q1) {
// 判断是否到达终点
if (deads.contains(cur))
continue;
if (q2.contains(cur))
return step;
visited.add(cur);
// 将一个节点的未遍历相邻节点加入集合
for (int j = 0; j < 4; j++) {
String up = plusOne(cur, j);
if (!visited.contains(up))
temp.add(up);
String down = minusOne(cur, j);
if (!visited.contains(down))
temp.add(down);
}
}
// 在这里增加步数
step++;
// temp 相当于 q1
// 这里交换 q1 q2,下一轮 while 就是扩散 q2
q1 = q2;
q2 = temp;
}
return -1;
}
/**
* 将cur[j]向上拨动一次
*
* @param cur 当前字符串
* @param j 需要向上拨动一次的位置
* @return 旋转后的字符串
*/
private String plusOne(String cur, int j) {
char[] ch = cur.toCharArray();
if (ch[j] == '9') {
ch[j] = '0';
} else {
ch[j] += 1;
}
return new String(ch);
}
/**
* 将cur[j]向下拨动一次
*
* @param cur 当前字符串
* @param j 需要向下拨动一次的位置
* @return 旋转后的字符串
*/
private String minusOne(String cur, int j) {
char[] ch = cur.toCharArray();
if (ch[j] == '0') {
ch[j] = '9';
} else {
ch[j] -= 1;
}
return new String(ch);
}
}
public class Study0752 {
public static void main(String[] args) {
System.out.println(new Solution01().openLock(new String[]{"0201", "0101", "0102", "1212", "2002"}, new String("0202")));
System.out.println(new Solution02().openLock(new String[]{"0000"}, "8888"));
}
}
结果: