leetcode刷题（67）——79. 单词搜索

一、题目

给定一个二维网格和一个单词，找出该单词是否存在于网格中。

单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

示例:

board =
[
  ['A','B','C','E'],
  ['S','F','C','S'],
  ['A','D','E','E']
]

给定 word = "ABCCED", 返回 true
给定 word = "SEE", 返回 true
给定 word = "ABCB", 返回 false

提示：

board 和 word 中只包含大写和小写英文字母。
1 <= board.length <= 200
1 <= board[i].length <= 200
1 <= word.length <= 10^3

二、思路——深度优先搜索 + 回溯

整体思路：

• 首先遍历二维网格，找到给定单词的第一个字母在网格中的位置；
• 从该位置开始向上下左右四个方向搜索，匹配给定单词的第二个字母；
• 匹配到第二个字母后，继续搜索匹配第三个字母，直至匹配到整个单词，返回 true；否则返回 false。

注意两点：

• 向上下左右搜索时，要注意网格的边界，不能出界（先往哪个方向搜索都可以）；
• 由于同一个单元格内的字母不允许被重复使用，所以我们可以对匹配过的位置打上标记；

三、代码

class Solution {
    private int[][] directions = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};// 上、下、左、右
    private int row, col;
    private boolean hasFind;
    private boolean[][] marked;

    public boolean exist(char[][] board, String word) {
        row = board.length;
        col = board[0].length;
        hasFind = false;
        // 网格规模小于单词长度，false
        if(row * col < word.length())
            return false;
        marked = new boolean[row][col];
        char[] chars = word.toCharArray();
        for(int i = 0; i < row; i++){
            for(int j = 0; j < col; j++){
                if(board[i][j] == chars[0]){
                    backTrack(board, chars, 1, i, j);
                    if(hasFind)
                        return true;
                }
            }
        }
        return false;
    }

    // curIndex表示当前匹配单词的字母的索引
    public void backTrack(char[][] board, char[] word, int curIndex, int x, int y){
        if(hasFind)
            return;
        if(curIndex == word.length){
            hasFind = true;
            return;
        }
        marked[x][y] = true;
        for(int[] d : directions){
            int i = x + d[0];
            int j = y + d[1];
            if(inArea(i, j) && !marked[i][j] && board[i][j] == word[curIndex]){
                backTrack(board, word, curIndex + 1, i, j);
            }
        }
        marked[x][y] = false;
    }

    // 判断位置(x, y)是否在网格中
    public boolean inArea(int x, int y){
        return x >= 0 && x < row && y >= 0 && y < col;
    }
}