要求:
给定一个二维网格和一个单词,找出该单词是否存在于网格中。单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.
算法如下:
// 回溯法
// 时间复杂度: O(m*n*m*n)
// 空间复杂度: O(m*n)
class Solution {
private int m;
private int n;
private int[][] d = {{-1,0}, {0,1}, {1,0}, {0,-1}};
private boolean[][] visited;
public boolean exist(char[][] board, String word) {
if(board == null || word == null)
throw new IllegalArgumentException("board or word can not be null!");
m = board.length;
if(m == 0)
throw new IllegalArgumentException("board can not be empty.");
n = board[0].length;
if(n == 0)
throw new IllegalArgumentException("board can not be empty.");
visited = new boolean[m][n];
for(int i=0; i<m; i++)
for(int j=0; j<n; j++)
if(searchWord(board, word, 0, i, j))
return true;
return false;
}
// 从 boart[startx][starty] 处开始找 word[index...word.length())
private boolean searchWord(char[][] board, String word, int index,
int startx, int starty){
// 终止条件
if(index == word.length() - 1)
return board[startx][starty] == word.charAt(index);
if(board[startx][starty] == word.charAt(index)){
visited[startx][starty] = true;
// 从 startx,starty出发, 向四个方向寻找
for(int i=0; i<4; i++){
int newx = startx + d[i][0];
int newy = starty + d[i][1];
if( inArea(newx, newy) && !visited[newx][newy] &&
searchWord(board , word, index+1, newx, newy) )
return true;
}
visited[startx][starty] = false; // 回溯
}
return false;
}
// 判断是否越界
private boolean inArea(int x, int y){
return x >= 0 && x < m && y >= 0 && y < n;
}
}