题解:bfs之密码锁

题目描述
玛雅人有一种密码,如果字符串中出现连续的2012四个数字就能解开密码。给一个长度为N的字符串,(2=<N<=13)该字符串中只含有0,1,2三种数字,问这个字符串要移位几次才能解开密码,每次只能移动相邻的两个数字。例如02120经过一次移位,可以得到20120,01220,02210,02102,其中20120符合要求,因此输出为1.如果无论移位多少次都解不开密码,输出-1。

输入
第一行输入N,第二行输入N个数字,只包含0,1,2

输出
样例输入
5
02120
5
02120
样例输出
1
1
每次交换一对,判断,进栈

#include <cstdio>
#include <cstdlib>
#include <climits>
#include <queue>
#include <set>
#include <iostream>
#include <string>
using namespace std;
struct Record{
	string code;
	int moveCount;
};
bool contain2012(string x)
{
	for(int i=3;i<x.length();i++)
	 if(x[i-3]=='2'&&x[i-2]=='0'&&x[i-1]=='1'&&x[i]=='2')
	  return true;
	return false;
}
int main()
{
 int n;
 while(cin>>n)
 {
  queue<Record>q;
  set<string>s;
  Record r;
  cin>>r.code;
  if(contain2012(r.code))    //无需移动 
  {
  	cout<<"0\n";continue;
  }
  r.moveCount=0;q.push(r);s.insert(r.code);
  int times2=0,times1=0,times0=0;
  int flag=true;                          //BFS 
  while(!q.empty())
  {
  	r=q.front();q.pop();
  	for(int i=0;i+1<r.code.length();i++)
  	{
  	 Record tmp=r;
	 swap(tmp.code[i],tmp.code[i+1]);
	 tmp.moveCount++;
	 if(contain2012(tmp.code))
	 {
	 	cout<<tmp.moveCount<<endl;
	 	while(!q.empty()) q.pop();
	 	flag=false;break;
	 }
	 if(s.find(tmp.code)==s.end())//找不到 
	 {
	 	s.insert(tmp.code);
	 	q.push(tmp);
	 }
	}
  }
  if(flag) cout<<"-1\n";
 }	
 return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43540515/article/details/92068638