剑指 Offer 12. 矩阵中的路径
在做这道题的时候偶然在评论区看见有前辈提到了使用引用,因此在这里回顾一下引用;
题目:给定一个 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
思路:使用深度优先搜索来解决这个问题:通过递归,朝着某个方向进行检索,如果这条路走不通,就回溯到上一个结点,然后沿另一个方向继续检索,以此类推。
递归终止的条件:
- 全部匹配到了,就结束递归;
- 行或者列越界了,或者当前的元素和字符串中的元素不匹配,或者当前的元素已经被访问过了,返回false,终止递归;
代码如下:
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[0].size();j++)
{
if(DFS(board,word,i,j,0))
{
return true;
}
}
}
return false;
}
//i、j表示存放字母的矩阵的下标,k表示字符串word中第k个元素
// 注意这里使用引用作为参数,效率得到明显的提高;
bool DFS(vector<vector<char>>& board, string &word, int i, int j, int k)
{
//如果下表越界或者说矩阵中的元素和字符串word中的元素对应不相等,就返回flase;
if(i<0 || i>=board.size() || j<0 || j>=board[0].size() || board[i][j] != word[k])
{
return false;
}
//如果k == word.size()-1,就表示word中的元素都被匹配到了,就返回true;
if(k == word.size()-1)
{
return true;
}
//设置一个临时变量来记录当前矩阵的值;便于之后回置
char temp = board[i][j];
//当访问到这个元素时,就把这个元素置为空,标志这个元素被访问过了;不过要注意的是,如果当前元素的相邻元素中没有符合要求的,需要回溯的时候,还要把矩阵中的这个元素从空恢复成原来的值;这里也可以用一个bool类型的数组,用来标记元素是否被访问了;
board[i][j] = '\0';
//递归地遍历其上下左右四个元素,注意这里连接关系用的是或,只要有符合条件的就行;
bool result = DFS(board, word, i + 1, j, k + 1) || DFS(board, word, i - 1, j, k + 1) || DFS(board, word, i, j + 1, k + 1) || DFS(board, word, i , j - 1, k + 1);
//搜索结束后要把 board[i][j]置为原来的元素
board[i][j] = temp;
return result;
}
};
在DFS函数中,参数word使用引用和不使用引用的区别非常大:
- 使用引用时(bool DFS(vector<vector>& board, string &word, int i, int j, int k) ):16 ms 7.5 MB
- 不适用引用时(bool DFS(vector<vector>& board, string word, int i, int j, int k) ):348 ms 173.2 MB
这效果简直了~;
对引用的深入理解请移步:对引用的深入理解