【洛谷 P1101】单词方阵 题解（dfs深度优先搜索）

题目描述

给一 n×n 的字母方阵，内可能蕴含多个 yizhong 单词。单词在方阵中是沿着同一方向连续摆放的。摆放可沿着 88 个方向的任一方向，同一单词摆放时不再改变方向，单词与单词之间可以交叉，因此有可能共用字母。输出时，将不是单词的字母用 * 代替，以突出显示单词。

输入格式

第一行输入一个数 n。(7≤n≤100)。

第二行开始输入n×n 的字母矩阵。

输出格式

突出显示单词的 n×n 矩阵。

输入输出样例

输入 #1复制

7
aaaaaaa
aaaaaaa
aaaaaaa
aaaaaaa
aaaaaaa
aaaaaaa
aaaaaaa

输出 #1复制

*******
*******
*******
*******
*******
*******
*******

输入 #2复制

8
qyizhong
gydthkjy
nwidghji
orbzsfgz
hhgrhwth
zzzzzozo
iwdfrgng
yyyygggg

输出 #2复制

*yizhong
gy******
n*i*****
o**z****
h***h***
z****o**
i*****n*
y******g

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define Z cout<<"\n"
const int N = 110;
int n;
char a[N][N], b[N][N]; //a用于输入，b用于存储符合的字符
char key[] = "yizhong";
int dx[8] = { -1,-1,-1,0,0,1,1,1 }; //dx,dy分别对应八个方位
int dy[8] = { -1,0,1,-1,1,-1,0,1 };
bool dfs(int x, int y, int step, int i) //x,y对应当前位置，step代表访问到了第几步(即访问key)，i代表当前是延哪个方向
{
	bool flag = false; //flag用来判断是否符合yizhong
	if (a[x][y] == key[step]) 
	{
		if (step >= 6) //当走到最后一个位置时，代表前面都符合题目要求,此时只需要给b当前坐标赋上g
		{
			b[x][y] = key[6];
			return true; 
		}
		int xx = x + dx[i];
		int yy = y + dy[i];
		if (xx >= 0 && xx <= n && yy >= 0 && yy <= n) //判断坐标是否越界
		{
			if (dfs(xx, yy, step + 1, i)) //只有当访问到最后一步时才符合条件
			{
				flag = true;
			}
		}
	}
	if (flag) //此时当前方向的7个位置都已经满足条件，只需要回溯时等于原来的字符就可以
	{
		b[x][y] = a[x][y];
	}
	return flag;
}
signed main()
{
	IOS;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			cin >> a[i][j];
			b[i][j] = '*';
		}
	}
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			for (int k = 0; k < 8; k++) //每个位置都要判断八次
			{
				dfs(i, j, 0, k);
			}
		}
	}
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			cout << b[i][j];
		}
		Z;
	}
	return 0;
}

算法小白的刷题日记。