【双向BFS】蓝桥杯真题 青蛙跳杯子

问题描述

       X星球的流行宠物是青蛙，一般有两种颜色：白色和黑色。
　　X星球的居民喜欢把它们放在一排茶杯里，这样可以观察它们跳来跳去。
　　如下图，有一排杯子，左边的一个是空着的，右边的杯子，每个里边有一只青蛙。

      *WWWBBB
　　
　　其中，W字母表示白色青蛙，B表示黑色青蛙，表示空杯子。
　　
　　X星的青蛙很有些癖好，它们只做3个动作之一：
　　1. 跳到相邻的空杯子里。
　　2. 隔着1只其它的青蛙（随便什么颜色）跳到空杯子里。
　　3. 隔着2只其它的青蛙（随便什么颜色）跳到空杯子里。
　　
　　对于上图的局面，只要1步，就可跳成下图局面：
　　WWWBBB
　　
　　本题的任务就是已知初始局面，询问至少需要几步，才能跳成另一个目标局面。
　　
　　输入为2行，2个串，表示初始局面和目标局面。
　　输出要求为一个整数，表示至少需要多少步的青蛙跳。

样例输入
WWBB
WWBB
样例输出
2

样例输入
WWWBBB
BBBWWW
样例输出
10

数据规模和约定
　　我们约定，输入的串的长度不超过15
　　资源约定：峰值内存消耗（含虚拟机） < 256M， CPU消耗 < 1000ms

思路

双向bfs：同时从初始字符串和目标字符串开始宽搜，直到两边出现了相同的中间字符串，结果就是【从初始字符串->中间字符串的步数】+【从目标字符串->中间字符串的步数】+1

代码

#include<bits/stdc++.h>
using namespace std;
map<string, int> cnt; //cnt记录到达该字符串需要的步数 
map<string, int> vis; //vis记录字符串是否访问过，值为1表示从初始字符串演变而来，值为2表示由目标字符串演变而来 
queue<string> que;
int n;

int main(){
	string s, e;
	cin>>s>>e;
	vis[s] = 1;	//初始字符串阵营为1 
	vis[e] = 2; //目标字符串阵营为2 
	que.push(s);
	que.push(e);
	n = s.size();
	
	while(que.size()){
		string t = que.front();
		que.pop();
		
		for(int i=0; i<n; i++){
			if(t[i]=='*'){
				if(i-1>=0 && t[i-1]!='*'){
					string tmp = t;
					swap(tmp[i], tmp[i-1]);
					if(vis[tmp]==0){	//若该字符串状态为访问过 
						vis[tmp] = vis[t];	//标记为所演变而来的阵营 
						cnt[tmp] = cnt[t]+1; //步数+1 
						que.push(tmp);
					}
					else if(vis[tmp]!=vis[t]){	//若该字符串已经出现过，且属不同阵营， 则得出答案 
						printf("%d\n", cnt[tmp]+cnt[t]+1);
						return 0;
					}					
				}
				
				if(i+1<n && t[i+1]!='*'){
					string tmp = t;
					swap(tmp[i], tmp[i+1]);
					if(vis[tmp]==0){
						vis[tmp] = vis[t];
						cnt[tmp] = cnt[t]+1;
						que.push(tmp);
					}
					else if(vis[tmp]!=vis[t]){
						printf("%d\n", cnt[tmp]+cnt[t]+1);
						return 0;
					}						
				}
				
				if(i-2>=0 && t[i-2]!='*'){
					string tmp = t;
					swap(tmp[i], tmp[i-2]);
					if(vis[tmp]==0){
						vis[tmp] = vis[t];
						cnt[tmp] = cnt[t]+1;
						que.push(tmp);
					}	
					else if(vis[tmp]!=vis[t]){
						printf("%d\n", cnt[tmp]+cnt[t]+1);
						return 0;
					}					
				}
				
				if(i+2<n && t[i+2]!='*'){
					string tmp = t;
					swap(tmp[i], tmp[i+2]);
					if(vis[tmp]==0){
						vis[tmp] = vis[t];
						cnt[tmp] = cnt[t]+1;
						que.push(tmp);
					}	
					else if(vis[tmp]!=vis[t]){
						printf("%d\n", cnt[tmp]+cnt[t]+1);
						return 0;
					}					
				}
				
				if(i-3>=0 && t[i-3]!='*'){
					string tmp = t;
					swap(tmp[i], tmp[i-3]);
					if(vis[tmp]==0){
						vis[tmp] = vis[t];
						cnt[tmp] = cnt[t]+1;
						que.push(tmp);
					}	
					else if(vis[tmp]!=vis[t]){
						printf("%d\n", cnt[tmp]+cnt[t]+1);
						return 0;
					}					
				}
				
				if(i+3<n && t[i+3]!='*'){
					string tmp = t;
					swap(tmp[i], tmp[i+3]);
					if(vis[tmp]==0){
						vis[tmp] = vis[t];
						cnt[tmp] = cnt[t]+1;
						que.push(tmp);
					}	
					else if(vis[tmp]!=vis[t]){
						printf("%d\n", cnt[tmp]+cnt[t]+1);
						return 0;
					}					
				}
			}
		}
	}
	
	return 0;
}