第五章 深度与广度优先搜索(上)
一、深度优先搜索
例题:
寻找最短路径:
剑指 Offer 12. 矩阵中的路径
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。
[["a","b","c","e"],
["s","f","c","s"],
["a","d","e","e"]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
示例 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
关键步骤:
递推工作:
标记当前矩阵元素: 将 board [ i ] [ j ] 修改为 空字符 ‘’ ,代表此元素已访问过,防止之后搜索时重复访问。
搜索下一单元格: 朝当前元素的 上、下、左、右 四个方向开启下层递归,使用 或 连接 (代表只需找到一条可行路径就直接返回,不再做后续 DFS ),并记录结果至 res 。
还原当前矩阵元素: 将 board[ i ] [ j ] 元素还原至初始值,即 word[k] 。
class Solution {
int width,height;
string word;
vector<vector<char>> board;
public:
bool issafe(int x,int y){ //边界合法
if(x>=0 && x<height && y>=0 && y<width)
return true;
return false;
}
bool dfs(int x,int y,int state){
if(state==word.length()) return true;
if(!issafe(x,y) || board[x][y]!=word[state]) return false;
board[x][y] = '\0';
bool res = dfs(x-1,y,state+1) || dfs(x+1,y,state+1) || dfs(x,y-1,state+1) || dfs(x,y+1,state+1);
board[x][y] = word[state];
return res;
}
bool exist(vector<vector<char>>& board, string word) {
this->board = board;
this->width = board[0].size();
this->height = board.size();
this->word = word;
for(int i=0;i<height;i++)
for(int j=0;j<width;j++){
if(dfs(i,j,0)) return true;
}
return false;
}
};
剑指 Offer 13. 机器人的运动范围
class Solution {
int **board;
int k,m,n;
public:
bool isaccess(int m,int n){ //判断到达条件
if(m%10 + m/10 + n%10 + n/10 > k) return false;
else return true;
}
void dfs(int x,int y){
if(x<0 || x>=m || y<0 || y>=n) return; //空间不合法
if(board[x][y]!=0) return; //已经访问过
if(isaccess(x,y)){
board[x][y] = 1; //能够到达
cout<<"board["<<x<<"]["<<y<<"]=1"<<endl;
dfs(x-1,y); dfs(x+1,y); dfs(x,y-1); dfs(x,y+1);
}
else board[x][y] = -1; //不能到达
}
int movingCount(int m, int n, int k) {
this->k=k; this->m=m; this->n=n;
int total = 0;
board = new int* [m]; //为board开辟空间,1表示到达,0表示未访问,-1表示到达
for(int i = 0; i < m; i++){
board[i] = new int[n];
memset(board[i],0,n*sizeof(int));
}
dfs(0,0);
for(int i=0;i<m;i++)
for(int j=0;j<n;j++){
if(board[i][j]==1) total++;
}
return total;
}
};