单词矩阵【dfs】

回归到算法的学习

今天记录深度优先搜索的一道题目，来自洛谷试炼场普及组，深度优先搜索专题P1101。
题目链接：https://www.luogu.org/problem/P1101

题目描述

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

输入格式

第一行输入一个数n。(7≤n≤100)。
第二行开始输入n×n的字母矩阵。

输出格式

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

样例输入/输出

算法思路

基础型深搜题目，我们在矩阵中找到 y 之后，沿一个路径走一步如果找到 i 就沿当前路径一直搜下去。同时记录脚印，如果走到了最后一步，就把脚印记录到路径里。继续搜索其他方向，或下一起始坐标。

算法实现

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
struct node{
	int x;
	int y;
}path[1000];
//路径 
int vis[120][120];
//方阵 
char fz[120][120];
char rule[]="yizhong";
//八个方向 
int dir[][2]={{-1,-1},{-1,0},{-1,1},{0,-1},{0,1},{1,-1},{1,0},{1,1}};
//搜索原型 
void dfs(int i, int j, node p[], int k, int cur){
	//走到最后一步搜索完成 
	if(cur==7){
		for(int i=0; i<7; i++){
			vis[p[i].x][p[i].y]=1;	//路径 
		}
	}
	else{
		//方向向量迈进一步 
		int dx=i+dir[k][0];
		int dy=j+dir[k][1];
		if( cur==6 || fz[dx][dy]==rule[cur+1]){
			//记下这一步 
			p[cur].x=i;
			p[cur].y=j;
			dfs(dx, dy, p, k, cur+1);
		}	
	}
}
int main()
{
	int n;
	cin>>n;
	for(int i=0; i<n; i++)
		cin>>fz[i];
	//搜索开始 
	for(int i=0; i<n; i++){
		for(int j=0; j<n; j++){
			//找到 y 
			if(fz[i][j]=='y')
				for(int k=0; k<8; k++){
					int x=i+dir[k][0];
					int y=j+dir[k][1];
					if(fz[x][y]=='i')
					//找到 i 并沿该路径搜索 
						dfs(i, j, path, k, 0);
				}
		}
	}
	//按照路径输出 
	for(int i=0; i<n; i++){
		for(int j=0; j<n; j++){
			if(vis[i][j])
				cout<<fz[i][j];
			else
				cout<<'*';
		}
		cout<<endl;
	}
	return 0;
}

题目结束。

工欲善其事，必先利其器