Acwing_179八数码【A*算法】

题目链接: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;
}