剑指offer - 矩阵中的路径

发表于 | 更新于

题目

https://www.nowcoder.com/questionTerminal/c61c6999eecb4b8f88a98f66b273a3cc

题意

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则该路径不能再进入该格子。

例如矩阵

中包含一条字符串”bcced”的路径，但是矩阵中不包含”abcb”路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

题解

经典的DFS。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
class Solution {
private:
    vector< vector<char> > g;
    vector< vector<int> > vis;
    int dirx[4]={0,1,0,-1};
    int diry[4]={1,0,-1,0};
    int len;
public:
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        len = strlen(str);
        g.resize(rows,vector<char>(cols));
        vis.resize(rows,vector<int>(cols));
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                g[i][j]=matrix[i*cols+j];
            }
        }
        //枚举起点
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                if(dfs(i,j,str,0))
                    return true;
            }
        }
        return false;
    }
private:
    bool dfs(int x,int y,char * str,int pos){
        if(str[pos]!=g[x][y]) return false;
        if(pos==len-1) return true;
        vis[x][y]=1;
        for(int i=0;i<4;i++){
            int nx = x+dirx[i];
            int ny = y+diry[i];
            if(judgeIn(nx,ny)&&!vis[nx][ny]&&dfs(nx,ny,str,pos+1))
                return true;
        }
        vis[x][y]=0;
        return false;
    }
    bool judgeIn(int x,int y){
        return 0<=x&&x<g.size()&&0<=y&&y<g[0].size();
    }
};