LC 打开转盘锁(C++ 实现)

问题描述

你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有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”
输出: 1

示例2

输入: deadends = [“8888”], target = “0009”
输出：1

示例3

输入: deadends = [“8887”,“8889”,“8878”,“8898”,“8788”,“8988”,“7888”,“9888”], target = “8888”
输出：-1

示例4

输入: deadends = [“0000”], target = “8888”
输出：-1

题解

#include <iostream>
#include <queue>
#include <unordered_set>
using namespace std;

// 打开转盘锁
class Solution {
public:
#if 0 
	采用广度优先遍历思路，从起点出发，每一次迭代访问四个节点，
	当搜索到Target节点时停止迭代
#endif
	int openLock(vector<string>& deadends, string target) {
	    // 1.初始化，采用unordered_set 标记访问过的节点; 借助队列进行广度优先遍历
		unordered_set <string> vis;
		vis.insert(deadends.begin(), deadends.end());  // 将无法访问的节点放入集合中
		if (vis.count("0000")) return -1;

		int step = 0;
		queue<string> st;
		st.push("0000");
		// 2. 进入迭代循环，判断条件:队列是否为空，不为空继续迭代。直到最终全部遍历仍未找到目标，返回-1
		while (!st.empty()) {
			// 2.1 遍历队列中的元素，若存在目标节点，则返回迭代次数
			int size = st.size();
			for (int i = 0; i < size; ++i) {
				string stemp = st.front();
				st.pop();
				if (stemp == target) return step;

				// 2.1.2 否则将相邻未出队列节点依次入队 
				for (int j = 0; j < 4; j++) {
					for (int count = -1; count <= 1; count += 2) {
						char c = (stemp[j] - '0' + count + 10) % 10 + '0';
						string s = stemp;
						s[j] = c;
						// 判断节点是否已被访问过
						if (!vis.count(s)) {
							st.push(s);
							vis.emplace(s);
						}
					}
				}
			}
			// 迭代次数加1
			step++;
		}
		// 2.2 全部遍历，未找到目标节点，返回 -1
		return -1;
	}
};

用例测试

int main() {
	Solution solution;

	// ---- 示例1 ---- //
	vector<string> sample1 = {"0201", "0101", "0102", "1212", "2002"};
	string target1 = "0202";
	cout << "sample1 step = " << solution.openLock(sample1, target1) << endl;

	// ---- 示例2 ---- //
	vector<string> sample2 = { "0201", "0101", "0102", "1212", "2002" };
	string target2 = "0202";
	cout << "sample2 step = " << solution.openLock(sample2, target2) << endl;

	return 0;
}