洛谷 P1101 单词方阵

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

输入：
    8                     输出：
    qyizhong              *yizhong
    gydthkjy              gy******
    nwidghji              n*i*****
    orbzsfgz              o**z****
    hhgrhwth              h***h***
    zzzzzozo              z****o**
    iwdfrgng              i*****n*
    yyyygggg              y******g
输入输出格式
输入格式：
第一行输入一个数n。(7 <= n <= 100)。

第二行开始输n×n的字母矩阵。

输出格式：
突出显示单词n×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 "bits/stdc++.h"
 
using namespace std;

char **c,**visited,**flag;
int n,dir[8][2]={0,1,1,0,0,-1,-1,0,1,1,1,-1,-1,-1,-1,1};


void input(){
    cin >> n; 
    c = (char**)malloc(sizeof(char*)*n);
    visited = (char**)malloc(sizeof(char*)*n);
    flag = (char**)malloc(sizeof(char*)*n);
    for(int i = 0; i < n; i ++){
        c[i] = (char*)malloc(sizeof(char)*n);
        visited[i] = (char*)malloc(sizeof(char)*n);
        flag[i] = (char*)malloc(sizeof(char)*n);
        for(int j = 0; j < n; j ++){
            cin >> c[i][j];
            visited[i][j] = 0;
            flag[i][j] = 0;
        }
    }
}
void print(){
    for(int i = 0; i < n; i ++){
        for(int j = 0; j < n; j ++){
            if(flag[i][j])
                cout << c[i][j]; 
            else
                cout << "*";
        }
        cout << endl;
    }
    cout << endl;
}
void mark(){
    for(int i = 0; i < n; i ++)
        for(int j = 0; j < n; j ++)
            if(visited[i][j]) flag[i][j] = 1;
}
void f(int x, int y,int x1, int y1,char* word,int len,int index){
    if(index == len){
        // success 
        mark();
        return; 
    }
    if(x == n || y == n || x < 0 || y < 0){
        // over bound
        return;
    }
    if(visited[x][y]){
        // visited
        return;
    }
    visited[x][y] = 1;
    if(word[index] == c[x][y]){
        f(x+x1, y+y1, x1, y1, word, len, index+1);
    }
    
    visited[x][y] = 0;
}

int main(){
    input();
    char *word = "yizhong";
    for(int i = 0; i < n; i ++){
        for(int j = 0; j < n; j ++){
            for(int k = 0; k < 8; k ++){
                f(i, j,dir[k][0],dir[k][1],word,strlen(word),0);
            }
        }
    }
    print(); 
    return 0;
}