矩阵中的路径
描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 \begin{bmatrix} a & b & c &e \\ s & f & c & s \\ a & d & e& e\\ \end{bmatrix}\quad⎣⎡asabfdcceese⎦⎤ 矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
示例1
输入:
"ABCESFCSADEE",3,4,"ABCCED"
返回值:
true
示例2
输入:
"ABCESFCSADEE",3,4,"ABCB"
返回值:
false
import java.util.*; public class Solution { int[] dir = new int[]{-1,0,1,0,-1};//路径移动辅助数组 boolean[][] visited;//标记是否访问过 public boolean hasPath(char[] matrix, int rows, int cols, char[] str) { if(str == null) { return false; } visited = new boolean[rows][cols];//初始化全局标记数组 for(int i = 0;i < rows;i++){ for(int j = 0;j < cols;j++){ if(dfs(matrix,i,j,rows,cols,str,0)) return true;//遍历所有起点,找到一个则返回 } } return false; } public boolean dfs(char[] matrix, int x, int y,int rows, int cols, char[] str,int index){ if( x < 0 || x >= rows || y < 0 || y >= cols || visited[x][y] == true) return false;//判断角标越界和是否访问过 if(str[index] != matrix[x*cols+y]){//判断是否等于下一节点:注意下边是x*cols+y而不是x*y+y return false; } else{ visited[x][y] = true;//置已访问 if(index == str.length-1) return true;//成功找到的结束标志 for(int i=0;i<4;i++){ if(dfs(matrix,x+dir[i],y+dir[i+1],rows,cols,str,index+1)) return true;//向四个方向遍历,找到一个即成功 } visited[x][y] = false;//都不成功,则证明当前节点不应该被访问,便于后续路径使用(这样可以避免每次遍历都需要新的标记数组) return false; } } }