79单词搜索

题目描述

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

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

示例:

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;
    }