剑指offer---矩阵中的路径

题目:矩阵中的路径

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

1 class Solution {
2 public:
3     bool hasPath(char* matrix, int rows, int cols, char* str)
4     {
5     
6     }
7 };

解题代码:

 1 class Solution {
 2 public:
 3     bool hasPath(char* matrix, int rows, int cols, char* str) {
 4         if(matrix == nullptr || rows < 1 || cols < 1 || str == nullptr)
 5             return false;
 6 
 7         bool *visited = new bool[rows * cols];
 8         memset(visited, 0, rows * cols);
 9 
10         int pathLength = 0;
11         for(int row = 0; row < rows; row++){
12             for(int col = 0; col < cols; col++){
13                 if(hasPathCore(matrix, rows, cols, row, col, str, pathLength, visited))
14                     return true;
15             }
16         }
17         delete[] visited;
18         return false;
19     }
20 private:
21     bool hasPathCore(char* matrix, int rows, int cols, int row, int col,
22         char* str, int &pathLength, bool* visited){
23         // 到达字符串末尾
24         if(str[pathLength] == '\0')
25             return true;
26 
27         bool hasPath = false;
28         if(row >= 0 && col >= 0 && row < rows && col < cols
29             && matrix[row * cols + col]==str[pathLength] && !visited[row * cols + col]){
30             pathLength++;
31             visited[row * cols + col] = true;
32 
33             // 若4个相邻格子中有一个方向的元素和要寻找的元素相同,则hasPath=true
34             hasPath = hasPathCore(matrix, rows, cols, row + 1, col, str, pathLength, visited)
35                         || hasPathCore(matrix, rows, cols, row - 1, col, str, pathLength, visited)
36                         || hasPathCore(matrix, rows, cols, row, col + 1, str, pathLength, visited)
37                         || hasPathCore(matrix, rows, cols, row, col - 1, str, pathLength, visited);
38 
39             // 如果4个相邻的格子都没有匹配字符串中下标为pathLength+1的字符,则表明定位错误,
40             // 需要回到前一个字符pathLength-1,重新定位
41             if(!hasPath){
42                 pathLength--;
43                 visited[row * cols + col] = false;
44             }
45         }
46         return hasPath;
47     }
48 };

 

posted on 2018-10-25 22:16  wangzhch  阅读(403)  评论(0编辑  收藏  举报

导航