剑指 Offer 12. 矩阵中的路径、79. 单词搜索:回溯寻找矩阵中符合条件的路径
给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中,返回 true ;否则,返回 false 。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
例如,在下面的 3×4 的矩阵中包含单词 "ABCCED"(单词中的字母已标出)。
示例 1:
输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
输出:true
示例 2:
输入:board = [["a","b"],["c","d"]], word = "abcd"
输出:false
回溯解法:
总体思路:需要寻找的单词的起点有可能是矩阵中任一位置的元素,所以首先按顺序枚举单词 word 的起点,然后在矩阵中向不同方向前进,判断对应位置的字母是否为符合条件的 word 后面的字母,若符合条件则继续深度搜索直到矩阵中的路径长度与单词 word 的长度相同时返回 true,若不符合条件则直接结束当前搜索返回 false。
解决题目的关键为深度优先搜索函数:
若矩阵中当前位置 (x, y) 处的字母与我们需要的 word 中 u 位置的字母不同,则表示此路径不符合条件,直接返回 false, 从其他的起点再开始搜索;若当前已经搜索到了单词的最后一个位置,说明已经找到了符合条件的路径,返回 true。
下面就是 DFS 的处理过程,体现出回溯的思想:使用 temp 存放矩阵 (x, y) 位置的字母,再将 (x, y) 位置标记为 '.' 表示该路径中的 (x, y) 位置已经经过;从当前位置到下一位置有四种方向,需要依次判断,若下一位置坐标仍然在矩阵界内且未遍历过,继续深度搜索,进入 DFS 的递归过程;若有一种路径满足 word 的条件,将在递归过程中反馈 true,若正常结束 DFS,则没有找到符合条件的路径,最后返回 false,并注意将矩阵 (x, y) 位置元素还原为 temp 以避免对其他起点的 DFS 过程造成影响。
class Solution {
public:
bool exist(vector<vector<char>>& board, string word) {
for(int i = 0; i < board.size(); i++)
for(int j = 0; j < board[i].size(); j++)
if(dfs(board, word, 0, i, j))
return true;
return false;
}
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
bool dfs(vector<vector<char>>& board, string& word, int u, int x, int y){
if(board[x][y] != word[u])
return false;
if(u == word.size() - 1)
return true;
char temp = board[x][y];
board[x][y] = '.';
for(int i = 0; i < 4; i++){
int a = x + dx[i], b = y + dy[i];
if(a < 0 || a > board.size() - 1 || b < 0 || b > board[0].size() - 1 || board[a][b] == '.')
continue;
if(dfs(board, word, u + 1, a, b))
return true;
}
board[x][y] = temp;
return false;
}
};