题目链接:矩阵中的路径
题意:请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 $\begin{bmatrix} a & b & c &e \\ s & f & c & s \\ a & d & e& e\\ \end{bmatrix}\quad⎣⎡asabfdcceese⎦⎤ $矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
题解:一个标准的回溯问题。我们把矩阵的每个点作为起点去向四个方向递归,不满足条件的会回到上个状态。就是有点难的写。。这里用一维存二维的方法也值得借鉴。
代码:
1 class Solution { 2 public: 3 bool dfs(char* matrix, int rows, int cols,int x,int y,char* str,int len,bool* vis){ 4 int pos = x*cols + y; //二维转一维 5 if(x < 0 || y < 0 || x >= rows || y >= cols || 6 matrix[pos] != str[len] || vis[pos] == true) return false; 7 if(str[len+1]=='\0') return true; //路已经找完了 8 //递归 9 vis[pos] = true; 10 //四个方向深搜 11 if(dfs(matrix,rows,cols,x-1,y,str,len+1,vis)|| 12 dfs(matrix,rows,cols,x+1,y,str,len+1,vis)|| 13 dfs(matrix,rows,cols,x,y-1,str,len+1,vis)|| 14 dfs(matrix,rows,cols,x,y+1,str,len+1,vis)) 15 return true; 16 vis[pos] = false; 17 return false; 18 } 19 bool hasPath(char* matrix,int rows,int cols,char* str){ 20 if(matrix == NULL || rows <= 0 || cols <= 0 || str== NULL) return false; 21 22 bool* vis = new bool[rows*cols]; 23 memset(vis,false,rows*cols); 24 for(int i = 0 ; i < rows ; i++){ 25 for(int j = 0; j < cols; j++){ 26 if(dfs(matrix,rows,cols,i,j,str,0,vis)) return true; 27 } 28 } 29 return false; 30 } 31 32 33 };