剑指offer打卡|矩阵中的路径

题目描述

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

代码

回溯法，即暴力求解的升级版。回溯法非常适合多个步骤组成的问题，每个步骤有多个选项。且回溯适合用递归来写。

public class Solution {
    //输入的虽然是个数组，但可以根据后面的rows和cols参数来确定出一个矩阵
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str)
    {
        int flag[]=new int[matrix.length];
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                if(backtrack(matrix,rows,cols,i,j,str,0,flag)){
                    return true;
                }
            }
        }
        return false;
    }
    public boolean backtrack(char[] matrix,int rows,int cols,int i,int j,char[] str,int k,int[] flag){
        int index=i*cols+j;//确定索引位置
        if(i<0||i>=rows||j<0||j>=cols||matrix[index]!=str[k]||flag[index]==1){
            return false;
        }
        if(k==str.length-1){
            return true;
        }
        flag[index]=1;
        //上下左右走一遍
        if(backtrack(matrix,rows,cols,i-1,j,str,k+1,flag)||
           backtrack(matrix,rows,cols,i+1,j,str,k+1,flag)||
           backtrack(matrix,rows,cols,i,j-1,str,k+1,flag)||
           backtrack(matrix,rows,cols,i,j+1,str,k+1,flag)){
            return true;
        }
        flag[index]=0;
        return false;
    }
}