【剑指offer】Java实现-矩阵中的路径

题目描述

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

思路

首先找到str[0]字符所在位置
判断它的路径上（上下左右）是否有str[1]
以此类推，并用一个boolean数组来标识是否已经访问过某个点
如果一条路径到某个点行不通之后，就开始回溯

代码

public class Solution {
    //判断在一个矩阵中是否存在一条包含某字符串所有字符的路径
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str){
        //如果矩阵、字符串为空或长度为0，返回false
        if(matrix==null || str==null || matrix.length==0 || str.length==0)
            return false;
        int len=matrix.length;
        //定义一个boolean数组、判断当前字符是否被访问过
        boolean[] flags=new boolean[len];
        //如果能够在矩阵中找到一个字符与字符串第一个字符相等，则返回true
        for(int r=0;r<rows;++r){
            for(int c=0;c<cols;++c){
                if(matrix[r*cols+c]==str[0]){
                    if(hasPath(matrix,rows,cols,r,c,str,0,flags)){
                        return true;
                    }
                }
            }
        }
        return false;
    }
    //判断路径中是否存在下一个字符
    private boolean hasPath(char[] matrix,int rows,int cols,int r,int c,
                           char[] str,int index,boolean[] flags){
        //定义一个索引从0开始到等于字符串长度结束
        if(index==str.length)
            return true;
        //如果行列超出矩阵范围，返回false
        if(r>=rows || r<0 || c>=cols || c<0)
            return false;
        //如果当前字符被访问过，或矩阵元素不等于字符串当前元素，则返回false
        if((flags[r*cols+c]==true || matrix[r*cols+c]!=str[index]))
            return false;
        flags[r*cols+c]=true;
        //是否存在一条包含某字符串所有字符的路径=向上查找||向下查找||向右查找||向左查找
        boolean hp=hasPath(matrix,rows,cols,r+1,c,str,index+1,flags)||
            hasPath(matrix,rows,cols,r-1,c,str,index+1,flags)||
            hasPath(matrix,rows,cols,r,c+1,str,index+1,flags)||
            hasPath(matrix,rows,cols,r,c-1,str,index+1,flags);
        if(hp==true)
            return true;
        flags[r*cols+c]=false;
        return false;
    }

}

目录