洛谷P1101 单词方阵

洛谷P1101 单词方阵（我的算法学习之路)

标签：字符串 搜索

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int Maxn=110;
char a[Maxn][Maxn];
bool vis[Maxn][Maxn];
int xx[]={1,1,1,0,0,-1,-1,-1};
int yy[]={1,0,-1,1,-1,-1,1,0};
string yz="yizhong";
int n;
void f(int x,int y){
	for(int i=0;i<n;i++)
	{
		int flag=1;
		for(int j=0;j<=6;j++)
		{
			int dx=x+j*xx[i];
			int dy=y+j*yy[i];
			if(dx<1||dx>n||dy<1||dy>n||yz[j]!=a[dx][dy]){
				flag=0;
				break;
			}
		}
		if(flag==1){
			for(int j=0;j<=6;j++)
			{
				int dx=x+j*xx[i];
				int dy=y+j*yy[i];
				vis[dx][dy]=1;
			}
		}
	}
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			cin>>a[i][j];
		}
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(a[i][j]=='y'){
				f(i,j);
			}
		}
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(vis[i][j]!=1)
				cout<<"*";
			else
				cout<<a[i][j];
		}
		cout<<endl;
	}
	return 0;
}