题目:
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.
链接:https://leetcode-cn.com/problems/word-search
思路:
1、这道题可以用深度优先搜索dfs来进行解决
2、按照从左到右的顺序查找字符串word中的元素,找到后查找下一个元素的方向是上下左右四个方向,按照dfs的方式进行遍历
3、需要注意边界条件和查找跳出条件
C++代码:
class Solution {
public:
bool dfs(vector<vector<char>>& board, string word, int d, int x, int y){
// 边界条件,不满足即跳出
// 如果当前元素不是我们想要的元素,即跳出
if(x<0 || x>=board.size() || y<0 || y>=board[0].size() || board[x][y]!=word[d]){
return false;
}
// 字符串word中所有元素都被查找到,说明已经找到了,返回true
if(d==word.size()-1){
return true;
}
char curr = board[x][y];
board[x][y] = 0; // 将当前位置的值设置成0,避免查找到之前走过的路径
// 上下左右四个方向进行查找
if(dfs(board,word,d+1,x+1,y) || dfs(board,word,d+1,x-1,y) || dfs(board,word,d+1,x,y+1) || dfs(board,word,d+1,x,y-1)){
return true;
}
board[x][y] = curr; // 恢复当前位置的值
return false;
}
bool exist(vector<vector<char>>& board, string word) {
if(board.size()==0){
return false;
}
int m = board.size();
int n = board[0].size();
for (int i=0; i<m; i++){
for (int j=0; j<n; j++){
if (dfs(board, word, 0, i, j)){
return true;
}
}
}
return false;
}
};
Python代码:
class Solution(object):
def dfs(self, board, word, d, x, y):
# 边界条件,不满足即跳出
# 如果当前元素不是我们想要的元素,即跳出
if(x<0 or x>=len(board) or y<0 or y>=len(board[0]) or board[x][y]!=word[d]):
return False
# 字符串word中所有元素都被查找到,说明已经找到了,返回true
if(d==len(word)-1):
return True
curr = board[x][y]
board[x][y] = 0 # 将当前位置的值设置成0,避免查找到之前走过的路径
# 上下左右四个方向进行查找
if(self.dfs(board,word,d+1,x+1,y) or self.dfs(board,word,d+1,x-1,y) or self.dfs(board,word,d+1,x,y+1) or self.dfs(board,word,d+1,x,y-1)):
return True
board[x][y] = curr # 恢复当前位置的值
return False
def exist(self, board, word):
"""
:type board: List[List[str]]
:type word: str
:rtype: bool
"""
if (not board):
return False
m = len(board)
n = len(board[0])
for i in range(m):
for j in range(n):
if self.dfs(board, word, 0, i, j):
return True
return False