Word Search -- 回溯法--力扣101算法题解笔记

6.3Word Search -- 回溯法

题目描述

给定一个字母矩阵，所有的字母都与上下左右四个方向上的字母相连接。给定一个字符串，求字符串能不能在字母矩阵中寻找到。

输入输出样例

Input ：word = "ABCCED",

board = $\['A','B','C','E'$ , $'S','F','C','S'$ ,

'A','D','E','E'\]

Output：true

输入的是一个二维字符数组和一个字符串，输出的是一个bool值，表示字符串是否可以被寻找到

题解

大体的解决方法可以从左上角的'A'开始，我们可以先向右、再向下、再向左。找到连续的ABCCED

这个跟前面两个的排列组合问题又不太一样，这种题目不能去修改输出方式，而是要修改访问标记。在我们对任意的位置进行深度优先搜索的时候，要先标记当前位置为已访问，避免重复遍历。在所有可能的搜索都完成后，再改回当前位置未访问。防止干扰其他位置搜索到当前位置。使用回溯法的话，可以只对一个二维的访问矩阵进行修改，不用把每次的搜索状态作为一个新对象传入递归函数中。

cpp 复制代码

#include <vector>
#include <iostream>
using namespace std;

void backtracking(int i, int j, vector<vector<char>>& board,
    string word, bool& find, vector<vector<bool>>& visited, int pos) {
    if (i < 0 || i >= board.size() || j < 0 || j >= board[0].size()) {
        return;
    }
    if (visited[i][j] || find || board[i][j] != word[pos]) {
        return;
    }
    if (pos == word.size() - 1) {
        find = true;
        return;
    }
    visited[i][j] = true;   //修改当前节点状态
    //递归遍历子节点们
    backtracking(i + 1, j, board, word, find, visited, pos + 1);
    backtracking(i - 1, j, board, word, find, visited, pos + 1);
    backtracking(i, j + 1, board, word, find, visited, pos + 1);
    backtracking(i, j - 1, board, word, find, visited, pos + 1);
    visited[i][j] = false;  //回改当前节点状态
}

bool exist(vector<vector<char>>& board, string word) {
    if (board.empty())    return false;
    int m = board.size(), n = board[0].size();
    vector<vector<bool>> visited(m, vector<bool>(n, false));
    bool find = false;
    for (int i = 0; i < m; ++i) {
        for (int j = 0; j < n; ++j) {
            backtracking(i, j, board, word, find, visited, 0);
        }
    }
    return find;
}

int main() {
    vector<vector<char>> board = {
        {'A','B','C','E'},
        {'S','F','C','S'},
        {'A','D','E','E'}
    };
    string word = "ABCCED";
    cout << (exist(board, word) ? "true" : "false") << endl;
    return 0;
}