洛谷-1101 单词方阵

题目描述
给一n×n的字母方阵，内可能蕴含多个“yizhong”单词。单词在方阵中是沿着同一方向连续摆放的。摆放可沿着 8 个方向的任一方向，同一单词摆放时不再改变方向，单词与单词之间可以交叉,因此有可能共用字母。输出时，将不是单词的字母用代替，以突出显示单词。例如：
输入：
8 输出：
qyizhong yizhong
gydthkjy gy****
nwidghji ni*****
orbzsfgz oz***
hhgrhwth hh***
zzzzzozo zo**
iwdfrgng i*****n
yyyygggg y****g
输入输出格式
输入格式：
第一行输入一个数n。(7≤n≤100)。
第二行开始输入n×n的字母矩阵。
输出格式：
突出显示单词的n×n矩阵。

输入样例#1：
8
qyizhong
gydthkjy
nwidghji
orbzsfgz
hhgrhwth
zzzzzozo
iwdfrgng
yyyygggg

输出样例#1：
yizhong
gy*****
ni*****
oz***
hh***
zo**
i*****n
y****g

解释：枚举’y’然后扫描8个方向，如果可以就标记一下，最后把未标记的全都输出*，就可以了

#include<iostream>
using namespace std;
char str[103][103];
int n=0;
char *mu="yizhong";
int dir[8][2]={0,1,0,-1,1,0,-1,0,1,1,-1,1,1,-1,-1,-1};
bool vis[103][103]={0};
bool check(int x,int y){
    if(x<n&&x>=0&&y<n&&y>=0) return 1;
    return 0;
}
void ok(int x,int y){
    for(int i=0;i<8;i++){
        int sum=0;
        int dx=x,dy=y;
        for(int j=0;mu[j];j++){
            if(check(dx,dy)&&str[dx][dy]==mu[j]) sum++;
            else break;
            dx+=dir[i][0];
            dy+=dir[i][1];
        }
        if(sum==7){
            dx=x,dy=y;
            for(int j=0;mu[j];j++){
                vis[dx][dy]=1;
                dx+=dir[i][0];
                dy+=dir[i][1];
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=0;i<n;i++) cin>>str[i];
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(str[i][j]=='y'){
                ok(i,j);
            }
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(!vis[i][j]) cout<<'*';
            else cout<<str[i][j];
        }
        cout<<endl;
    }
    return 0;
}