我的代码
#include<bits/stdc++.h>
using namespace std;
int n;
string s1,s2;
vector<int> temp;
struct Node{
string s;
int ind;//空格位置
vector<int> vc;
};
void bfs(){
queue<Node> q;
q.push({s1,n,temp});
while(!q.empty()){
Node now=q.front();
temp=now.vc;
q.pop();
if(now.s==s2){//合法结束状态结束搜索
for(int i=0;i<temp.size();i++){
cout<<temp[i]+1;
if((i+1)%5!=0) cout<<" ";
else cout<<endl;
}
break;
}
if(now.ind-1>=0){//左边棋子和右边空格交换
if(now.s[now.ind-1]=='W'){
Node next=now;
swap(next.s[now.ind-1],next.s[now.ind]);
temp.push_back(now.ind-1);
q.push({next.s,now.ind-1,temp});
temp.pop_back();
}
}
if(now.ind-2>=0&&now.s[now.ind-1]=='B'&&now.s[now.ind-2]=='W'){//左边棋子跳到右边空格
Node next=now;
swap(next.s[now.ind-2],next.s[now.ind]);
temp.push_back(now.ind-2);
q.push({next.s,now.ind-2,temp});
temp.pop_back();
}
if(now.ind+1<2*n+1){//右边棋子和左边空格交换
if(now.s[now.ind+1]=='B'){
Node next=now;
swap(next.s[now.ind+1],next.s[now.ind]);
temp.push_back(now.ind+1);
q.push({next.s,now.ind+1,temp});
temp.pop_back();
}
}
if(now.ind+2<2*n&&now.s[now.ind+1]=='W'&&now.s[now.ind+2]=='B'){//右边棋子跳到左边空格
Node next=now;
swap(next.s[now.ind+2],next.s[now.ind]);
temp.push_back(now.ind+2);
q.push({next.s,now.ind+2,temp});
temp.pop_back();
}
}
}
int main(){
cin>>n;
for(int i=0;i<n;i++){
s1+='W';
s2+='B';
}
s1+=' ';
s2+=' ';
for(int i=n+1;i<2*n+1;i++){
s1+='B';
s2+='W';
}
bfs();
return 0;
}
题解
这道题目和前面的搜索方法一样,到达每个状态之后有四个方向可以走,然后依次往下搜索就可以了。
这道题目难在两个地方。一是怎么记录我们已经到达过的状态,这里我们采用map来记录我们到达过的状态。二是怎么保存我们我们走过的路径,这里我们使用vector来记录。
1
#include <iostream>
2
#include <map>
3
#include <queue>
4
#include <vector>
5
#include <cstring>
6
#include <algorithm>
7
#include <cstdio>
8
using namespace std;
9
string s = "";
10
string ans = "";
11
map<string, bool> mp;
12
int dx[] = {-2, -1, 1, 2};// 可以移动的位置
13
int n, x;
14
15
struct node {
16
string s; // 模拟移动的字符串
17
vector<int> res; // 记录空格移动的位置
18
node(){}
19
node(string _s, vector<int> _res) {
20
s = _s;
21
res = _res;
22
}
23
};
24
25
vector<int> bfs() {
26
vector<int> v;
27
queue<node> q;
28
q.push(node(s, v));
29
mp[s] = 1;
30
while(!q.empty()) {
31
s = q.front().s;
32
v = q.front().res;
33
q.pop();
34
x = v.size() == 0 ? n : v[v.size() - 1]; // 得到空格最后的位置
35
if (s == ans) { // bfs 的出口
36
return v;
37
}
38
for (int i = 0; i < 4; ++i) {
39
int u = x + dx[i];
40
if (u >= 0 && u < n * 2 +1) {
41
swap(s[x], s[u]);
42
if (mp.find(s) == mp.end()) { // 查找这个状态是否记录过
43
mp[s] = 1; // 把这个状态标记
44
v.push_back(u);
45
q.push(node(s, v));
46
v.pop_back();
47
}
48
swap(s[x], s[u]);
49
}
50
}
51
}
52
}
53
54
int main() {
55
freopen("chess.in", "r", stdin);
56
freopen("chess.out", "w", stdout);
57
cin >>n;
58
for (int i = 0; i < n; ++i) {
59
s += 'w';
60
ans += 'b';
61
}
62
s += ' ' + ans;
63
ans = s;
64
reverse(ans.begin(), ans.end());
65
vector<int> res = bfs();
66
for (int i = 0; i < res.size(); ++i) {
67
if (i % 5 == 4) {
68
cout << res[i] + 1 << endl;
69
}
70
else {
71
cout << res[i] + 1 << ' ';
72
}
73
}
74
}