矩阵中的路径(剑指Offer-12)

题面

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。例如，在下面的3×4的矩阵中包含一条字符串“bfce”的路径（路径中的字母用加粗标出）。

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

但矩阵中不包含字符串“abfb”的路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入这个格子。

示例

示例 1：

1输入：board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
2输出：true

示例 2：

1输入：board = [["a","b"],["c","d"]], word = "abcd"
2输出：false

限制

  • 1 <= board.length <= 200
  • 1 <= board[i].length <= 200

思路

dfs即可。

代码

 1class Solution {
 2public:
 3    bool exist(vector<vector<char>>& board, string word) {
 4        n = board.size();
 5        if(n <= 0) return false;
 6
 7        m = board[0].size();
 8        if(m <= 0) return false;
 9
10        for(int i=0; i<n; i++){
11            for(int j=0; j<m; j++){
12                if(dfs(board, word, i, j, 0))
13                    return true;
14            }
15        }
16        return false;
17    }
18private:
19    int n, m;
20    bool dfs(vector<vector<char>>& board, string word, int i, int j, int k){
21        if(i<0 || i>=n || j<0 || j>=m || board[i][j]!=word[k])  return false;
22        if(k == word.size()-1)  return true;
23
24        board[i][j] = '\0';
25        bool ans = dfs(board, word, i+1, j, k+1) || dfs(board, word, i, j+1, k+1) ||
26                   dfs(board, word, i-1, j, k+1) || dfs(board, word, i, j-1, k+1);
27        board[i][j] = word[k];
28        return ans;
29    }
30};