问题描述
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;
}