luogu P1101 单词方阵

https://www.luogu.org/problem/P1101

题目描述

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

输入：
    8                     输出：
    qyizhong              *yizhong
    gydthkjy              gy******
    nwidghji              n*i*****
    orbzsfgz              o**z****
    hhgrhwth              h***h***
    zzzzzozo              z****o**
    iwdfrgng              i*****n*
    yyyygggg              y******g

输入格式

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

第二行开始输入$n\times n$的字母矩阵。

输出格式

突出显示单词的$n\times 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<algorithm>
#include<iostream>
#include<stdio.h>
using namespace std;
int n;
string c="yizhong";
char a[101][101],res[101][101];
int next[8][2]={{1,1},{1,0},{1,-1},{0,1},{0,-1},{-1,0},{-1,1},{-1,-1}};
//方向数组，全局变量
void dfs(int x,int y)
{
	for(int i=0; i<8; i++) //搜索8个方向
	{
		int b=1;
		for(int j=1; j<=6; j++)//j和上面每个要找的字符串的每个字母对应
		{
			int nx=x+j*next[i][0];
			int ny=y+j*next[i][1];
			
			if(nx<1||nx>n||ny<1||ny>n)//检查是否越界
			{
				b=0;
				break;
			}
			
			if(c[j]!=a[nx][ny]) //检查对应位置上的字母
			{
				b=0;
				break;
			}
		}
		
		if(b==0)//如果此方向没找到该字符串，则继续寻找其他方向
			continue;
		
		for(int j=0; j<=6; j++)
		{
			int nx=x+j*next[i][0];
			int ny=y+j*next[i][1];
			res[nx][ny]=a[nx][ny];//在res数组相应位置上标记上该字母
		}
	}
	
	return;
}

int main()
{
	scanf("%d",&n);
	for(int i=1; i<=n; i++)
		for(int j=1; j<=n; j++)
			scanf(" %c",&a[i][j]);
	
	for(int i=1; i<=n; i++)
		for(int j=1; j<=n; j++)
			if(a[i][j]=='y') //找到 y 
				dfs(i,j);
	
	for(int i=1; i<=n; i++)
	{
		for(int j=1; j<=n; j++)
		{
			if(res[i][j]==0) //没有标记字母的输出 
				res[i][j]='*';
			
			printf("%c",res[i][j]);
		}
		printf("\n");
	}
	
	return 0; 
}