剑指 Offer -- 矩阵中的路径(六十五)

剑指 Offer – 矩阵中的路径(六十五)

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

代码(牛客上已 AC)

使用 search(mat, i, j, start, str) 判断当前 mat[i][j] 是否等于 str[start], 如果相等, 那么就进行 DFS 再依次判断 str[start + 1] 之后的字符是否存在于 mat 中. 只有当最后访问到 str 的末尾, 即 str[start] == '\0' 时, 才能返回 true.
另外, 由于访问过的位置不能再访问了, 因此呢, 当然可以引入额外的空间 seen[i][j] 判断 (i, j) 是否已经被访问, 然而这里用 mat[i][j] = '#' 表示 (i, j) 已经被访问了, 但别忘记之后要用 tmp 将 mat[i][j] 给恢复出来.

class Solution {
public:
    bool hasPath(char* matrix, int rows, int cols, char* str) {
        m = rows, n = cols;
        vector<vector<char>> mat(m, vector<char>(n));
        for (int i = 0; i < m * n; ++i) mat[i / n][i % n] = matrix[i];
        for (int i = 0; i < m; ++i)
            for (int j = 0; j < n; ++j)
                if (search(mat, i, j, 0, str)) return true;
        return false;
    }
private:
    int m, n;
    bool search(vector<vector<char>> &mat, int i, int j, int start, char *str) {
        if (str[start] == '\0') return true;
        if (i < 0 || i >= m || j < 0 || j >= n
            || mat[i][j] == '#'
            || mat[i][j] != str[start])
            return false;
        char tmp = mat[i][j];
        mat[i][j] = '#';

        bool res = search(mat, i + 1, j, start + 1, str) ||
            search(mat, i - 1, j, start + 1, str) ||
            search(mat, i, j + 1, start + 1, str) ||
            search(mat, i, j - 1, start + 1, str);
        mat[i][j] = tmp;
        return res;
    } 
};