题目描述
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
[‘A’,‘B’,‘C’,‘E’],
[‘S’,‘F’,‘C’,‘S’],
[‘A’,‘D’,‘E’,‘E’]
]给定 word = “ABCCED”, 返回 true.
给定 word = “SEE”, 返回 true.
给定 word = “ABCB”, 返回 false.
思路分析
回溯算法。用一个相同大下的数组存储是否访问过当前元素。(也可以用原数组做存储,减少空间复杂度)
- 路径:已经访问过的字符。
- 选择列表:当前能访问的字符。
- 结束条件:超出边界或者字符串全部匹配访问到。
代码实现
public boolean exist(char[][] board, String word) {
if (word == null || board.length == 0 || board[0].length == 0) {
return false;
}
char[] chars = word.toCharArray();
int row = board.length;
int col = board[0].length;
int[][] memo = new int[row][col];
for (int i = 0; i < row; i++) {
for (int j = 0; j < col; j++) {
//先找到第一个匹配,开始后续找
if (process(board, chars, 0, i, j, row, col, memo)) {
return true;
}
}
}
return false;
}
public boolean process(char[][] board, char[] chars, int index, int i, int j, int row, int col, int[][] memo) {
if (index >= chars.length) {
return true;
}
if (i < 0 || i >= row || j < 0 || j >= col || memo[i][j] != 0 || chars[index] != board[i][j]) {
return false;
}
memo[i][j] = 1;
boolean result = process(board, chars, index + 1, i - 1, j, row, col, memo) ||
process(board, chars, index + 1, i + 1, j, row, col, memo) ||
process(board, chars, index + 1, i, j - 1, row, col, memo) ||
process(board, chars, index + 1, i, j + 1, row, col, memo);
memo[i][j] = 0;
return result;
}