《剑指 Offer》——65、矩阵中的路径

1. 本题知识点

回溯法

2. 题目描述

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

例如：
$$\left[\begin{array}{cccc}a& b& c& e\\ s& f& c& s\\ a& d& e& e\end{array}\right]$$
矩阵中包含一条字符串 “bcced” 的路径，但是矩阵中不包含 “abcb” 路径，因为字符串的第一个字符 b 占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

3. 解题思路

回溯法

使用回溯法进行求解，当搜索到某一格子时，若满足条件，继续向四个方向搜索；若不满足条件，则退回到上一个格子，重新选择要搜索的格子。（注意：已经搜索过的格子要做个标记，退回到上一个格子时，还要清除标记）

4. 代码

public class Solution {
    /**
     * 判断在一个矩阵中是否存在一条包含某字符串所有字符的路径，
     * 如果存在则返回 true，不存在则返回 false
     *
     * @param matrix 矩阵的一维数组
     * @param rows   矩阵的行数
     * @param cols   矩阵的列数
     * @param str    要查找的字符串
     * @return
     */
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str) {
        // 将矩阵的一维数组转换为二维数组
        char[][] realMatrix = createMatrix(matrix, rows, cols);
        // 创建标记数组，用来标记搜索过的位置
        boolean marked[][] = new boolean[rows][cols];
        // 从矩阵的每一个位置出发一次
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                if (backtracking(realMatrix, i, j, str, 0, marked)) {
                    return true;
                }
            }
        }
        return false;
    }

    /**
     * 判断从一个出发点开始，在一个矩阵中是否存在一条包含某字符串所有字符的路径
     * 如果存在则返回 true，不存在则返回 false
     *
     * @param realMatrix 矩阵的二维数组
     * @param i          出发点的行下标
     * @param j          出发点的列下标
     * @param str        要查找的字符串
     * @param index      当前查找字符对应字符串中的下标
     * @param marked     标记数组，用来标记搜索过的位置
     * @return
     */
    private boolean backtracking(char[][] realMatrix, int i, int j, char[] str, int index, boolean[][] marked) {
        // 如果搜索到字符串所有字符，就返回 true
        if (index == str.length) {
            return true;
        }
        // 如果当前出发点在矩阵以外，则返回 false
        if (i > realMatrix.length - 1 || i < 0 || j > realMatrix[0].length - 1 || j < 0) {
            return false;
        }
        // 如果当前出发点不匹配当前查找字符，返回 false
        if (realMatrix[i][j] != str[index]) {
            return false;
        }
        // 如果当前出发点被标记，返回 false
        if (marked[i][j]) {
            return false;
        }
        // 标记当前出发点
        marked[i][j] = true;
        // 如果当前出发点匹配当前查找字符成功，则向其他四个方向递归调用该函数
        if (realMatrix[i][j] == str[index++]) {
            // 如果有任意方向搜索到字符串所有字符，返回 true
            if (backtracking(realMatrix, i - 1, j, str, index, marked)
                    || backtracking(realMatrix, i + 1, j, str, index, marked)
                    || backtracking(realMatrix, i, j - 1, str, index, marked)
                    || backtracking(realMatrix, i, j + 1, str, index, marked)) {
                return true;
            }
            // 否则，清除标记，返回 false
            else {
                marked[i][j] = false;
                return false;
            }
        }
        return false;
    }


    /**
     * 将矩阵的一维数组转换为二维数组
     *
     * @param matrix 矩阵的一维数组
     * @param rows   矩阵的行数
     * @param cols   矩阵的列数
     * @return
     */
    private char[][] createMatrix(char[] matrix, int rows, int cols) {
        // 创建矩阵的二维数组
        char[][] realMatrix = new char[rows][cols];
        // 赋值矩阵的二维数组
        for (int i = 0, index = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                realMatrix[i][j] = matrix[index++];
            }
        }
        return realMatrix;
    }
}