(剪枝？回溯)剑指 矩阵中的路径

题目描述

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

运行时间：17ms

占用内存：9292k

 

public class Solution {
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str)
    {
        //一个用来记录是否已走过的数组
        boolean[] visited = new boolean[matrix.length];
        //找到矩阵中某个格子，该格子的值等于字符串的第一个字符，作为入口；
        for(int i=0; i<matrix.length; i++){
            if(matrix[i] == str[0]){
                boolean res = helper(matrix, i, visited, rows, cols, str, 0);
                if(res)
                    return true;
            }
        }
        return false;
    }
    
    private boolean helper(char[] matrix, int mi, boolean[] v, int rows, int cols, char[] str, int si){
        //越界， 当前格子已经被访问过， 当前格子的值和字符串待匹配的值不相等
        if(mi < 0 || mi >= matrix.length || v[mi] || matrix[mi] != str[si])
            return false;
        v[mi] = true;
        // 到这一步已经满足了matix[mi] == str[si]
        if(si == str.length - 1) 
            return true;
        //判断上下左右是否能走得通，只要有能走通的就可以返回true，不必再回溯其他方向
        if(helper(matrix, mi-cols, v, rows, cols, str, si+1)
          || helper(matrix, mi+cols, v, rows, cols, str, si+1)
          || helper(matrix, mi-1, v, rows, cols, str, si+1)
          || helper(matrix, mi+1, v, rows, cols, str, si+1))
            return true;
        //这个格子走不通， 不必访问
        v[mi] = false;
        return false;
    }
}