LeetCode 79.單詞搜索 - JavaScript

題目描述：給定一個二維網格和一個單詞，找出該單詞是否存在於網格中。

單詞必須按照字母順序，經過相鄰的單元格內的字母構成，其中「相鄰」單元格是那些水平相鄰或垂直相鄰的單元格。同一個單元格內的字母不容許被重複使用。

例子：

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

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

解法：深度優先遍歷 + 回溯

準備兩個函數exist() 和 __exist()。

exist() 用於循環遍歷網格，當前元素等於單詞的第一個字母時，進入 __exist() 函數。代碼實現以下：

// ac地址：https://leetcode-cn.com/problems/word-search/
// 原文地址：https://xxoo521.com/2020-02-17-word-search/

/**
 * @param {character[][]} board
 * @param {string} word
 * @return {boolean}
 */
var exist = function(board, word) {
    const rowNum = board.length;
    const colNum = board[0].length;

    for (let i = 0; i < rowNum; ++i) {
        for (let j = 0; j < colNum; ++j) {
            if (board[i][j] === word[0]) {
                const isExist = __exist(board, word, i, j, {});
                if (isExist) return true; // 找到就返回
            }
        }
    }
    return false;
};

__exist() 函數是從指定位置開始，按照「上下左右」四個方向遞歸遍歷，直到找到/找不到全部匹配的字母爲止。代碼實現以下：

/**
 * @param {character[][]} board
 * @param {string} word
 * @param {number} row
 * @param {number} col
 * @param {object} visited
 * @return {boolean}
 */
function __exist(board, word, row, col, visited) {
    // 單詞中字母所有匹配，說明能夠搜索到，返回true
    if (!word.length) {
        return true;
    }

    const key = `${row}-${col}`;
    // 越界、以前訪問過、單詞首字母和當前元素不相同，返回false
    if (
        row >= board.length ||
        row < 0 ||
        col >= board[0].length ||
        col < 0 ||
        visited[key] ||
        board[row][col] !== word[0]
    ) {
        return false;
    }

    visited[key] = true;
    word = word.slice(1);
    // 下、上、右、左搜索（順序不重要）
    const success =
        __exist(board, word, row + 1, col, visited) ||
        __exist(board, word, row - 1, col, visited) ||
        __exist(board, word, row, col + 1, visited) ||
        __exist(board, word, row, col - 1, visited);

    // success爲false時，就是回溯
    visited[key] = success;
    return success;
}

代碼中若是 success 爲 false，說明以當前元素爲出發點，搜索剩餘字母失敗。那麼應該將visited[key]重置爲 false，以方面其餘路徑使用。

例如對於如下數組，要搜索abbcbd。按照代碼裏的方向搜索邏輯，會先找到 abbd，而後發現查找失敗，此時就要回溯。不然當按照正確方向找來時，visited 中的值是錯誤的。

a b b
d b c

更多資料

• 📖Blog：劍指 Offer 題解 + JS 代碼
• 🐱Github ：https://github.com/dongyuanxin/blog
• 🌟 公衆號：心譚博客