用 Go 劍指 Offer 12. 矩陣中的路徑 (DFS + 回溯)

2023-04-10 18:00:28

給定一個 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
 

提示:

m == board.length
n = board[i].length
1 <= m, n <= 6
1 <= word.length <= 15
board 和 word 僅由大小寫英文字母組成

來源:力扣(LeetCode)
連結:https://leetcode.cn/problems/ju-zhen-zhong-de-lu-jing-lcof
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

dfs + 回溯 解法


func exist(board [][]byte, word string) bool {
   wordb := []byte(word) // 轉化一下 string 方便遍歷
    rows := len(board)
    cols := len(board[0])

    for i := 0; i < rows; i++ {
        for j := 0; j < cols; j++ {
            if(dfs(board, wordb, i, j, 0)) { // 呼叫 dfs 函數
                return true
            }
        }
    }
    return false
}

func dfs(board [][]byte, wordb []byte, i int, j int, k int) bool {
    rows := len(board)
    cols := len(board[0])
//先考慮 false 情況:越界、不相等、重複
if i >= rows || i < 0 || j >= cols || j < 0 || board[i][j] != wordb[k] { return false }
// 遍歷完成返回true
if k == len(wordb) - 1 { return true } //歸零標記已遍歷 byte 型別的0值為 0x0 board[i][j] = 0x0

//四個方向遍歷 res := dfs(board, wordb, i + 1, j, k + 1) || dfs(board, wordb, i - 1, j , k + 1) || dfs(board, wordb, i, j + 1, k + 1) || dfs(board, wordb, i, j - 1, k + 1) board[i][j] = wordb[k] return res }