剑指offer(矩阵中的路径)

矩阵中的路径

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

这一道题用回溯的思想去做

先贴一下代码

public class Solution {
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str){
        boolean [] visitFlags = new boolean [matrix.length];
        for(int row = 0; row < rows ; row++){
            for(int col = 0; col < cols; col++){
                if(helper(matrix , rows , cols , row , col , 0 , str , visitFlags)){
                    return true;
                }
            }
        }
        return false;
    }
    
    public boolean helper(char[] matrix , int rows , int cols , int row , int col , int k , char[] str , boolean [] visitFlags){
        int index = row * cols + col;
        
        if(row < 0 || col < 0 || row >= rows || col >= cols || visitFlags[index] || matrix[index] != str[k]){
            return false;
        }
        visitFlags[index] = true;
        
        if(str.length - 1 == k){
            return true;
        }
        
        k++;
        
        if(helper(matrix , rows , cols , row - 1 , col , k , str , visitFlags)
          || helper(matrix , rows , cols , row + 1 , col , k , str , visitFlags)
          || helper(matrix , rows , cols , row , col -1 , k , str , visitFlags)
          || helper(matrix , rows , cols , row , col + 1 , k , str , visitFlags)){
            return true;
        }
        
        visitFlags[index] = false;
        
        return false;
    }
    


}

因为matrix是一维的,所以按照index查询的时候,注意index的取值

main函数遍历每一个位置,然后helper函数的作用是查看当前查到的字符是否满足要求。

猜你喜欢

转载自blog.csdn.net/weixin_41999706/article/details/89214263