[Java]剑指offer:矩阵中的路径

题目描述

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

 

示例1 输入

[[a,b,c,e],[s,f,c,s],[a,d,e,e]],"abcced"

返回值

true

示例2 输入

[[a,b,c,e],[s,f,c,s],[a,d,e,e]],"abcb"

返回值

false

思路

对矩阵中的每个位置使用深度优先搜索以查找是否有符合的字符路径

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param matrix char字符型二维数组 
     * @param word string字符串 
     * @return bool布尔型
     */
    
    boolean[][] visited;
    int[][] directions = {{-1,0},{0,-1},{0,1},{1,0}};
    int m,n;
    int len;
    public boolean hasPath (char[][] matrix, String word) {
        // write code here
        m = matrix.length;
        n = matrix[0].length;
        len = word.length();
        visited = new boolean[m][n];
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                //遍历整个矩阵 把每个位置都作为起点试一次
                if(dfs(matrix,word,i,j,0)){
                    return true;
                }
            }
        }
        return false;
    }
    
    public boolean dfs(char[][] matrix,String word,int i,int j,int index){
        //最后一个字符
        if(index==len-1){
            return matrix[i][j] == word.charAt(index);
        }
        if(matrix[i][j]==word.charAt(index)){
            visited[i][j] = true;
            // 遍历i、j的上下左右位置
            for(int k=0;k<4;k++){
                int x = i + directions[k][0];
                int y = j + directions[k][1];
                if(inArea(x,y)&&!visited[x][y]){
                    if(dfs(matrix,word,x,y,index+1)){
                        return true;
                    }
                }
            }
            visited[i][j] = false;
        }
        return false;
    }
    
    public boolean inArea(int x,int y){
        return x>=0&&x<m&&y>=0&&y<n;
    }
}