[洛谷：单词方阵]DFS深搜题解

戳我：洛谷普及组P1101

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

#include <iostream>
using namespace std;
int dir[][2]={ {-1,0},{-1,1},{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1} };  //定义8个方向向量
char pattern[10]="yizhong";  //匹配模式串
int n,book[105][105];  //标记数组
int u,ans[14];  //用以暂时存储答案的ans数组及其控制下表变量
char a[105][105];
void dfs(int x,int y,int di,int cnt) {
    if(cnt==7) {  //cnt==7 返回true说明模式串已经全部匹配完毕，那么这个时候就可以对标记数组进行标记了
        for(int i=0;i<13;i+=2) {
            book[ ans[i] ][ ans[i+1] ]=1;
        }
        u=0;  //由于u是全局变量，return之前要记得清零
        return ;
    }
    if(x<1||y<1||x>n||y>n) {
        u=0;  //如果当前模式串还没匹配完成，而坐标已经超出表范围了，那么应该return（注：此选择分支结构必须放在上面选择分支结构之后）
        return ;
    }
    if(a[x][y]==pattern[cnt]) {  
        ans[u++]=x;
        ans[u++]=y;  //如果与当前模式串对应字符匹配，存入备选答案数组
        int tx=x+dir[di][0];
        int ty=y+dir[di][1];
        dfs(tx,ty,di,cnt+1);
    }
    else {  //如果不匹配，886...
        u=0;  //由于u是全局变量，return之前要记得清零
        return ;
    }
}
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++) {
            for(int k=0;k<8;k++) {
                dfs(i,j,k,0);  //对每一个点都进行一次DFS深搜，但是要注意方向每次都应该固定下来
            }
        }
    }
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            if(book[i][j])  cout<<a[i][j];
            else            cout<<'*';
        }
        cout<<endl;
    }
    return 0;
}