剑指 offer 刷题——12. 矩阵中的路径

一、题目

二、思路

回溯法：

• 创建一个和字符矩阵大小相同的布尔型矩阵，记录矩阵中的位置是否走过，走过为 true，没走过则为false；
• 从 (0, 0) 开始，遍历字符矩阵，先找到字符串的第一个字符在字符矩阵中的位置；
• 从该位置向左、右、上、下四个方向探索，匹配字符串的下一个字符；
• 若全部匹配成功，返回 true；否则，回溯至上一步，继续探索，直至遍历完整个字符数组；
• 探索过程中，要注意边界条件，行和列不能越界。

三、代码

public class Solution {
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str){
        boolean[] flag = new boolean[matrix.length];
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                if(backTrack(matrix, i, j, 0, rows, cols, str, flag))
                    return true;
            }
        }
        return false;
    }
    
    /**
    i：行坐标
    j：列坐标
    k：字符串的索引，表示当前匹配到第 k 位字符
    rows：行数
    cols：列数
    str：要匹配的字符串
    flag：标志矩阵
    */
    public boolean backTrack(char[] matrix, int i, int j, int k, int rows, int cols, char[] str, boolean[] flag){
        // 计算匹配到的第一个字符在矩阵中的位置
        int index = i * cols + j;
        // 递归结束条件
        if(i < 0 || i >= rows || j < 0 || j >= cols || matrix[index] != str[k] || flag[index])
            return false;
        // 当 k = str.length - 1 时，说明整个字符串都匹配到了
        if(k == str.length - 1)
            return true;
        flag[index] = true;
        // 向上下左右四个方向搜索，匹配下一个字符
        if(backTrack(matrix, i - 1, j, k + 1, rows, cols, str, flag) || // 向上
              backTrack(matrix, i + 1, j, k + 1, rows, cols, str, flag) || // 向下
              backTrack(matrix, i, j - 1, k + 1, rows, cols, str, flag) || // 向左
              backTrack(matrix, i, j + 1, k + 1, rows, cols, str, flag)){ // 向右
            return true;
        }
        // 没有匹配到，回溯
        flag[index] = false;
        return false;
    }
}