给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.
这个题就是一个迷宫题,一般就是使用DFS来进行搜索,不同的是这里不是一个开始点,要对每一个点都开始一次搜索。用一个数组来标记所有的地方是否被走过。具体思路看程序会更清楚一些。有意思的一点是,C++程序中如果使用bool类型来定义标记数组,那么就会很慢而且占用空间非常大,而使用int类型定义标记数组时,程序运行速度会很快而且占用空间会减少很多。
C++源代码:
class Solution {
public:
bool exist(vector<vector<char>>& board, string word) {
if(board.empty() || word.empty()) return false;
int m = board.size(), n = board[0].size();
vector<vector<int>> flag(m, vector<int>(n, 0));
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(search(board, word, 0, i, j, flag))
return true;
}
}
return false;
}
bool search(vector<vector<char>>& board, string& word, int idx, int x, int y, vector<vector<int>>& flag){
if(idx == word.size()) return true;
int m = board.size(), n = board[0].size();
if(x<0 || y<0 || x>=m || y>=n || flag[x][y] || board[x][y]!=word[idx]) return false;
flag[x][y] = 1;
bool res = search(board, word, idx+1, x+1, y, flag) ||
search(board, word, idx+1, x-1, y, flag) ||
search(board, word, idx+1, x, y+1, flag) ||
search(board, word, idx+1, x, y-1, flag);
flag[x][y] = 0;
return res;
}
};
python3源代码:
class Solution:
def exist(self, board, word):
"""
:type board: List[List[str]]
:type word: str
:rtype: bool
"""
if len(board)==0 or len(word)==0:
return False
m = len(board)
n = len(board[0])
flag = [[0 for j in range(n)] for i in range(m)]
for i in range(m):
for j in range(n):
if self.search(board, word, 0, i, j, flag):
return True
return False
def search(self, board, word, idx, x, y, flag):
if idx == len(word):
return True
m = len(board)
n = len(board[0])
if x<0 or y<0 or x>=m or y>=n or flag[x][y] or board[x][y]!=word[idx]:
return False
flag[x][y] = 1
res = self.search(board, word, idx+1, x+1, y, flag) or self.search(board, word, idx+1, x-1, y, flag) or self.search(board, word, idx+1, x, y+1, flag) or self.search(board, word, idx+1, x, y-1, flag)
flag[x][y] = 0
return res