题目描述
玛雅人有一种密码,如果字符串中出现连续的2012四个数字就能解开密码。给一个长度为N的字符串,(2=<N<=13)该字符串中只含有0,1,2三种数字,问这个字符串要移位几次才能解开密码,每次只能移动相邻的两个数字。例如02120经过一次移位,可以得到20120,01220,02210,02102,其中20120符合要求,因此输出为1.如果无论移位多少次都解不开密码,输出-1。
输入描述
输入包含多组测试数据,每组测试数据由两行组成。 第一行为一个整数N,代表字符串的长度(2<=N<=13)。 第二行为一个仅由0、1、2组成的,长度为N的字符串。
输出描述
对于每组测试数据,若可以解出密码,输出最少的移位次数;否则输出-1。
输入
5 02120
输出
1
思路
典型的BFS,节点就是一个字符串,与它相邻的节点就是由它交换一次所得的字符串。
从初始节点出发,看看它交换一次形成的字符串里哪些没出现过,这些字符串如果有符合要求的就结束BFS返回结果,如果不符合要求那就把它们加入队列。遍历完之后还是没碰到符合要求的字符串就返回-1。
#include <stdio.h>
#include <iostream>
#include <queue>
#include <map>
#include <string>
using namespace std;
queue<string> Q;
map<string, int> M;
//判断str是否含有2012
bool Judge(string str) {
if (str.find("2012") == string::npos) {
return false;
} else {
return true;
}
}
string Swap(string s, int i) {
char temp = s[i];
s[i] = s[i+1];
s[i+1] = temp;
return s;
}
//广度优先算法
int BFS(string str) {
M.clear();
while (!Q.empty()) {
Q.pop();
}
Q.push(str);
M[str] = 0;
//队列不为空时
while (!Q.empty()) {
str = Q.front();
Q.pop();
for (int i = 0; i < str.length()-1; i++) {
string newStr = Swap(str, i);
if (M.find(newStr) == M.end()) { //如果M中没有key为newStr的值
M[newStr] = M[str]+1;
if (Judge(newStr)) {
return M[newStr];
}
Q.push(newStr);
}
}
}
return -1;
}
int main() {
int n;
string s;
while (cin >> n) {
cin >> s;
if (Judge(s)) {
cout << 0 << endl;
} else {
cout << BFS(s) << endl;
}
}
return 0;
}