【剑指offer】矩阵中的路径

题目描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则之后不能再次进入这个格子。 例如 $a$ $b$ $c$ $e$ $s$ $f$ $c$ $s$ $a$ $d$ $e$ $e$ 这样的$3X4$ 矩阵中包含一条字符串"$bcced$“的路径，但是矩阵中不包含”$abcb$"路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

思路：先将所有的点转换为矩阵；然后就是熟悉的$dfs$进行搜索。

class Solution {
public:
    char maze[100][100];
    bool vis[100][100];
    int m, n;
    //四个方向
    int dir[4][2] = {{0, -1}, {0, 1}, {-1, 0}, {1, 0}};
    //建立矩阵
    void build(char* matrix, int rows, int cols) {
        int cnt = 0;
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                maze[i][j] = matrix[cnt++];
            }
        }
    }
    //dfs遍历
    bool dfs(int x, int y, char* str, int cnt, int length) {
        if (cnt >= length)
            return true;
        //标记该点
        vis[x][y] = true;
        for (int i = 0; i < 4; ++i) {
            int tx = x + dir[i][0];
            int ty = y + dir[i][1];
            //判断是否符合要求
            if (vis[tx][ty] == 0 && tx >= 0 && tx < m && ty >= 0
               && ty < n && str[cnt] == maze[tx][ty]) {
                if (dfs(tx, ty, str, cnt + 1, length))
                    return true;
            }
        }
        return false;
    }
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        m = rows, n = cols;
        build(matrix, rows, cols);
        memset(vis, 0, sizeof(vis));
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
            	//找到出发点，遍历
                if (maze[i][j] == str[0] && dfs(i, j, str, 1, strlen(str))) {
                    return true;
                }
                //回溯
                memset(vis, 0, sizeof(vis));
            }
        }
        return false;
    }
};