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