玛雅人的密码

题目描述

玛雅人有一种密码，如果字符串中出现连续的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;
}