题目链接:Acwing_179八数码
代码片段里有详细注释,其中具体的算法含义请参考OI WiKi这个网站
AC代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<unordered_map>
#include<vector>
#define PIS pair<int,string>
using namespace std;
const int N = 1010;
int f(string s)
{ //估值函数
//此处设计的估值为 当前状态各个数字到终态的对应数字处的曼哈顿距离的总和
//即当前状态的 1 到终态的 1 的曼哈顿距离 加上 当前状态的 2 到终态的 2 的曼哈顿距离 加上......
int ans = 0;
for (int i = 0; i < s.size(); i++)
{
if (s[i] != 'x')
{
int t = s[i] - '1';
ans += abs(i / 3 - t / 3) + abs(i % 3 - t % 3);
}
}
return ans; //返回估值
}
unordered_map < string, int > d; //用来记录到达某个状态所需的步数
unordered_map < string, pair<char, string> > pre; //用来记录到达某状态的前驱状态,三个维度分别表示:当前状态、从前驱状态转移到当前状态所移动的方向、前驱状态
priority_queue < PIS, vector <PIS>, greater<PIS> > heap; //小根堆:first存储起点到终点的总估计值,second存储当前状态
string bfs(string st)
{
string en = "12345678x"; //定义终态
char op[] = "urdl"; //定义方向
d[st] = 0; //将起点到起点的距离设为 0 。
heap.push({ f(st), st }); //将初态入队
int fx[4] = { -1, 0, 1, 0 }, fy[4] = { 0, 1, 0, -1 };
while (heap.size())
{ //堆未空
PIS t = heap.top(); //取出堆顶
heap.pop();
string state = t.second;
if (state == en)break; //如果终点出队,则结束循环
int x, y; //找出空格x的位置
for (int i = 0; i < 9; i++)
{
if (state[i] == 'x')
{
x = i / 3;
y = i % 3;
break;
}
}
string s = state; //将前驱状态保存,方便还原
for (int i = 0; i < 4; i++)
{
int a = x + fx[i], b = y + fy[i];
if (a < 0 || a > 2 || b < 0 || b > 2) continue; //判断移动位置是否在矩阵内
state = s; //还原状态
swap(state[x * 3 + y], state[a * 3 + b]); //移动空格
if (!d.count(state) || d[state] > d[s] + 1)
{ //如果当前状态未出现过 或者 当前状态的步数可以更新
d[state] = d[s] + 1; //更新步数
pre[state] = { op[i], s }; //储存前驱
heap.push({ d[state] + f(state), state }); //新状态入队
}
}
}
string ans;
while (en != st)
{ //求出移动方案
ans += pre[en].first;
en = pre[en].second; //从终态往回求移动方案直到终态等于初态
}
reverse(ans.begin(), ans.end()); //由于是从终态开始的,所以方案是倒着的,需要翻转一下。
return ans; //返回方案
}
int main()
{
string st, seq;
string c;
for (int i = 1; i <= 9; i++)
{
cin >> c; //小技巧:让你输入字符时用字符串代替,可以自动滤去空格和回车
st += c[0]; //记录开始时的矩阵
if (c[0] != 'x')seq += c[0]; //记录下所有的数字用来计算逆序对
}
//计算逆序对对数
int cn = 0;
for (int i = 0; i < 8; i++)
for (int j = i; j < 8; j++)
if (seq[i] > seq[j])cn++; //如果前面的数字大于后面的数字,则将个数加一。
if (cn & 1)printf("unsolvable"); //如果逆序对个数为奇数,则无解。
else cout << bfs(st); //否则进行 BFS
return 0;
}