题目描述:
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。
[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
思路:
递归,深度优先搜索(dfs)
有四个方向可以选择
代码如下:
class Solution {
public:
bool backtrack(vector<vector<char>>& board, string &word,int i,int j,int flag){
if(flag==word.size()) return true;
if(i<0||i>=board.size()||j<0||j>=board[0].size()) return false;
if(board[i][j]!=word[flag]) return false;
board[i][j]='#';
if(backtrack(board,word,i+1,j,flag+1)||backtrack(board,word,i,j+1,flag+1)||backtrack(board,word,i-1,j,flag+1)||backtrack(board,word,i,j-1,flag+1))
return true;
board[i][j]=word[flag];
return false;
}
bool exist(vector<vector<char>>& board, string word) {
if(word.size()<1) return false;
if(board.empty()||board[0].empty()){
return word.empty();
}
for(int i=0;i<board.size();i++){
for(int j=0;j<board[0].size();j++){
if(backtrack(board,word,i,j,0))
return true;
}
}
return false;
}
};
Python:
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
def dfs(i,j,k):
if not 0<= i <len(board) or not 0<= j <len(board[0]) or board[i][j] != word[k]:return False
if k==len(word)-1: return True
temp=board[i][j]
board[i][j]='#'
res=dfs(i+1,j,k+1) or dfs(i,j+1,k+1) or dfs(i-1,j,k+1) or dfs(i,j-1,k+1)
board[i][j]=temp
return res
for i in range(len(board)):
for j in range(len(board[0])):
if dfs(i,j,0):
return True
return False