题目描述
X星球的流行宠物是青蛙,一般有两种颜色:白色和黑色。
X星球的居民喜欢把它们放在一排茶杯里,这样可以观察它们跳来跳去。
如下图,有一排杯子,左边的一个是空着的,右边的杯子,每个里边有一只青蛙。
*WWWBBB
其中,W字母表示白色青蛙,B表示黑色青蛙,*表示空杯子。
X星的青蛙很有些癖好,它们只做3个动作之一:
1. 跳到相邻的空杯子里。
2. 隔着1只其它的青蛙(随便什么颜色)跳到空杯子里。
3. 隔着2只其它的青蛙(随便什么颜色)跳到空杯子里。
对于上图的局面,只要1步,就可跳成下图局面:
WWW*BBB
本题的任务就是已知初始局面,询问至少需要几步,才能跳成另一个目标局面。
输入为2行,2个串,表示初始局面和目标局面。
输出要求为一个整数,表示至少需要多少步的青蛙跳。
输入
输入为2行,2个串,表示初始局面和目标局面。
输出
输出要求为一个整数,表示至少需要多少步的青蛙跳。
样例输入
*WWBB
WWBB*
样例输出
2
#include<iostream>
#include<algorithm>
#include<queue>
#include<string>
#include<map>
using namespace std;
struct node{
string s;
int idx;
int step;
};
map<string,bool> vis;
int dx[6] = {-1,1,-2,2,-3,3};
string s,tar;
void bfs(int x){
node now,next;
now.idx = x,now.step = 0,now.s = s;
queue<node> q;
q.push(now);
vis[now.s] = 1;
while(q.size()){
now = q.front();
q.pop();
if(now.s == tar){
cout<<now.step<<endl;
return;
}
for(int i = 0;i < 6;i ++){
next.idx = now.idx + dx[i];
if(next.idx >= 0 && next.idx < s.length()){
next.s = now.s,next.step = now.step + 1;
swap(next.s[now.idx],next.s[next.idx]);
if(!vis[next.s]){
q.push(next);
vis[next.s] = 1;
}
}
}
}
}
int main()
{
cin>>s>>tar;
int idx = s.find('*');
bfs(idx);
return 0;
}
JAVA解法:
import java.util.*;
public class Main {
public static StringBuilder sb = new StringBuilder("");
public static HashMap<String,Boolean> vis = new HashMap<>();
public static String begin,end;
public static int len;
public static int []dx = {-1,1,-2,2,-3,3};
public static StringBuilder swap(StringBuilder t,int p1,int p2) {
char c = t.charAt(p1);
t.setCharAt(p1, t.charAt(p2));
t.setCharAt(p2, c);
return t;
}
public static int bfs(int idx) {
Queue<State> q = new LinkedList<>();
q.add(new State(begin,idx,0));
vis.put(begin, true);
while(q.size() != 0) {
State cur = q.poll();
if(cur.s.equals(end)) {
return cur.step;
}
for(int i = 0;i < 6;i ++) {
int a = cur.idx + dx[i];
if(a >= 0 && a < len) {
StringBuilder t = new StringBuilder(cur.s);
String s = swap(t,cur.idx,a).toString();
if(!vis.containsKey(s)) {
q.add(new State(s,a,cur.step+1));
vis.put(s, true);
}
}
}
}
return 1;
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
begin = sc.next();
end = sc.next();
len = begin.length();
int idx = begin.indexOf('*');
for(int i = 0;i < begin.length();i ++) {
sb.append(begin.charAt(i));
}
System.out.println(bfs(idx));
}
}
class State{
String s;
int idx;
int step;
State(String s,int idx,int step){
this.s = s;
this.idx = idx;
this.step = step;
}
}