C++回溯的方法解决问题
需要创建一个记录数组来记录回溯过程中访问的状态,防止回溯过程中重复访问某个元素
需要注意的是,回溯返回时,要对状态值重置
这道题的搜索路径是上下左右四个方向,并且可以从矩阵中的任一个位置开始回溯,所以要对矩阵中的每一个位置进行回溯,如果回溯中遇到了契合的字符串,则直接返回true
#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param matrix char字符型vector<vector<>>
* @param word string字符串
* @return bool布尔型
*/
void dfs(vector<vector<char>> &matrix, vector<vector<bool>> &flag, string curword, int i, int j) {
// 可向上下左右四个方向去探索
// count用于计数,判断当前字符串加到哪里了
if (i < 0 || i > matrix.size() - 1 || j < 0 || j > matrix[0].size() - 1) return;
if (!flag[i][j]) return;
if (matrix[i][j] == target[curword.size()]) {
curword += matrix[i][j];
flag[i][j] = false;
}
else return;
if (curword.size() == target.size()) {
haspath = true;
return;
} else {
dfs (matrix, flag, curword, i+1, j);
if (haspath) return;
dfs (matrix, flag, curword, i-1, j);
if (haspath) return;
dfs (matrix, flag, curword, i, j-1);
if (haspath) return;
dfs (matrix, flag, curword, i, j+1);
}
flag[i][j] = true; // 回溯,回来后要记得修改状态
return;
}
bool hasPath(vector<vector<char> >& matrix, string word) {
// write code here
// 判断矩阵中是否有一个路径是否和word相同
// 理应用回溯法来做
target = word;
flag = vector<vector<bool>>(matrix.size(), vector<bool>(matrix[0].size(), true));
for (int i = 0; i < matrix.size(); i++) {
for (int j = 0; j < matrix[0].size(); j++) {
// 对矩阵中的每一个元素,每一个元素都可作为起点
dfs (matrix, flag, "",i,j);
if (haspath) return true;
}
}
return haspath;
}
private:
bool haspath = false;
vector<vector<bool>> flag; // 先定义类型,但并不初始化
string target;
};

京公网安备 11010502036488号