[LeetCode]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

解题思路

详细思路请参考 面试题12. 矩阵中的路径

代码

class Solution {
    public boolean exist(char[][] board, String word) {
        int m = board.length;
        int n = board[0].length;
        boolean[][] isVisited = new boolean[m][n];
        char[] chs = word.toCharArray();
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if(board[i][j] == chs[0]){
                    if(dfs(board, i, j, isVisited, chs, 0)){
                        return true;
                    }
                }
            }
        }
        return false;
    }

    private boolean dfs(char[][] board, int i, int j, boolean[][] isVisited, char[] chs, int step){
        if(step == chs.length){
            return true;
        }
        if(i<0 || i>= board.length || j<0 || j>= board[0].length || isVisited[i][j] || board[i][j] != chs[step]){
            return false;
        }
        isVisited[i][j] = true;
        int[][] direction = {{-1, 0}, {0, -1}, {0, 1}, {1, 0}};
        for(int k=0; k<4; k++){
            if(dfs(board, i+direction[k][0], j+direction[k][1], isVisited, chs, step+1)){
                return true;
            }
        }
        isVisited[i][j] = false;
        return false;
    }
}