问题描述
给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中,返回 true ;否则,返回 false 。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
输入示例
board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
输出示例
true
解题思路
使用回溯的思想,设函数 check(i, j, k) 表示判断以网格的 (i, j) 位置出发,能否搜索到单词 word[k…],其中 word[k…] 表示字符串 word 从第 k 个字符开始的后缀子串。如果能搜索到,则返回 true,反之返回 false。函数 check(i, j, k) 的执行步骤如下:
这样,我们对每一个位置 (i, j) 都调用 check(i, j, 0) 进行检查:只要有一处返回 true,就说明网格中能够找到相应的单词,否则说明不能找到。
为了防止重复遍历相同的位置,需要额外维护一个与 board 相同大小的 visited 数组,用于标识每个位置是否被访问过。每次遍历时,需要跳过已经被访问过的位置。
解题代码
class Solution {
public boolean exist(char[][] board, String word) {
int h = board.length;
int w = board[0].length;
boolean[][] visited = new boolean[h][w];
for(int i = 0; i < h; i++) {
for(int j = 0; j < w; j++) {
boolean flag = check(board, visited, i, j, word, 0);
if(flag) {
return true;
}
}
}
return false;
}
public boolean check(char[][] board, boolean[][] visited, int i, int j, String s, int k) {
// 当前位置不匹配,直接返回false
if(board[i][j] != s.charAt(k)) {
return false;
// 根据长度判断全匹配,直接返回true
} else if(k == s.length() - 1) {
return true;
}
// 标识当前可以匹配到当前字符
visited[i][j] = true;
// 方向
int[][] directions = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
boolean result = false;
for(int[] dir : directions) {
int newi = i + dir[0];
int newj = j + dir[1];
if(newi >= 0 && newi < board.length && newj >= 0 && newj < board[0].length) {
if(!visited[newi][newj]) {
boolean flag = check(board, visited, newi, newj, s, k+1);
if(flag) {
result = true;
break;
}
}
}
}
visited[i][j] = false;
return result;
}
}