剑指 Offer 12. 矩阵中的路径

题目

给定一个m x n 二维字符网格board 和一个字符串单词word 。如果word 存在于网格中，返回 true ；否则，返回 false 。

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

例如，在下面的 3×4 的矩阵中包含单词 “ABCCED”（单词中的字母已标出）。

示例 1：输入：board = [[“A”,”B”,”C”,”E”],[“S”,”F”,”C”,”S”],[“A”,”D”,”E”,”E”]], word = “ABCCED”
输出：true

示例 2：
输入：board = [[“a”,”b”],[“c”,”d”]], word = “abcd”
输出：false

提示：
1 <= board.length <= 200
1 <= board[i].length <= 200
board 和 word 仅由大小写英文字母组成

注意：本题与主站79题相同

题解

public boolean exist(char[][] board, String word) {
    int rows = board.length, columns = board[0].length, len = word.length();
    // 访问过的路径
    boolean[][] visited = new boolean[rows][columns];

    TrFunction<Integer, Integer, Integer, Boolean> backtrack = new TrFunction<Integer, Integer, Integer, Boolean>() {
        @Override
        public Boolean apply(Integer row, Integer column, Integer index) {
            // 已经匹配最后一个字符
            if (index == len) {
                return true;
            }

            // 越界或者已经访问过或者字符不匹配
            if (row < 0 || row >= rows
                || column < 0 || column >= columns
                || visited[row][column]
                || board[row][column] != word.charAt(index)) {
                return false;
            }

            visited[row][column] = true;
            // 继续匹配下一个字符
            if (this.apply(row + 1, column, index + 1)
                || this.apply(row, column + 1, index + 1)
                || this.apply(row - 1, column, index + 1)
                || this.apply(row, column - 1, index + 1)) {
                return true;
            }
            // 回溯
            visited[row][column] = false;

            return false;
        }
    };

    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < columns; j++) {
            if (backtrack.apply(i, j, 0)) {
                return true;
            }
        }
    }

    return false;
}

interface TrFunction<P1, P2, P3, R> {
    R apply(P1 p1, P2 p2, P3 p3);
}