【BFS】玛雅人的密码

时间限制：1秒 空间限制：65536K 热度指数：5412

 算法知识视频讲解

校招时部分企业笔试将禁止编程题跳出页面，为提前适应，练习时请使用在线自测，而非本地IDE。

题目描述

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

示例1

输入

5

02120

输出

1

1. 典型的BFS题目，节点为字符串

2. 巧妙地运用map来判断搜索树中是否有重复的节点

3. s.find("2012")!=string::npos 匹配字符串中的模式

#include<bits/stdc++.h>
using namespace std;
bool judge(string s){
    if(s.find("2012")!=string::npos) return true;
    return false;
}
string mySwap(string s, int i){
    char c = s[i+1];
    s[i+1] = s[i];
    s[i] = c;
    return s;
}
int main(){
    int n;
    while(cin >> n){
        string s;
        bool flag = false;
        queue<string> q;
        map<string, int> M;
        cin >> s;
        if(judge(s)){
            cout << 0 << endl;
            continue;
        }
        q.push(s);
        M[s] = 0;
        while(!q.empty()){
            string curS = q.front();
            q.pop();
            for(int i = 0; i < curS.length()-1; i++){
                string newS = mySwap(curS, i);
                if(M.count(newS) == 0){
                    M[newS] = M[curS]+1;
                    if(judge(newS)){
                        cout << M[newS] << endl;
                        flag = true;
                        break;
                    }
                    else{
                        M[newS] = M[curS]+1;
                        q.push(newS);
                    }
                }
            }
            if(flag) break;
        }
        if(!flag){
            cout << -1 << endl;
        }
    }

    return 0;
}