洛谷p1101 dfs(八方向搜索）

P1101 单词方阵

#include<bits/stdc++.h>
using namespace std;
const string str="yizhong";
char a[110][110]; int c[110][110];
int n,dx[8]={1,1,-1,-1,0,1,0,-1},dy[8]={1,-1,1,-1,1,0,-1,0};


void dfs(int x,int y)
{
    for(int i=0;i<8;i++)
    {
        int fg=1;
        for(int j=1;j<7;j++)
        {
            int z=x+dx[i]*j,m=y+dy[i]*j;
            if(z>n||m>n||z<1||m<1) {fg=0;break;}
            if(a[z][m]!=str[j]) {fg=0;break;}
        }
        if(fg)
        {
            for(int j=0;j<7;j++)
            {
            int z=x+dx[i]*j,m=y+dy[i]*j;
            c[z][m]=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') dfs(i,j);
    
    for(int i=1;i<=n;i++) 
    {
        for(int j=1;j<=n;j++)
        {
            if(c[i][j]) cout<<a[i][j];
            else cout<<"*";
        }
        cout<<endl;
    }
        
}