剑指Offer12. 矩阵中的路径Golang版

剑指Offer12. 矩阵中的路径Golang版

1. 问题描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。例如，在下面的3×4的矩阵中包含一条字符串“bfce”的路径（路径中的字母用加粗标出）。

[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]

但矩阵中不包含字符串“abfb”的路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入这个格子。

示例 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

思路

回溯法
回溯法模板：

void dfs ( int x ) {
	if ( 到达目的地 ) {
		输出解
		return
	}
	
	for （ i = 1; i <= 方案数; i++ ）{
		if ( 方案可行 ) {
			保存路径
			dfs( x + 1 )
			保存回复前的状态
		}
	}
}

代码

func exist(board [][]byte, word string) bool {
    visit := make([][]bool, len(board))
    for i := 0; i < len(visit); i++ {
        visit[i] = make([]bool,len(board[0]))
    }

    for j := 0; j < len(board); j++ {
        for k := 0; k < len(board[0]); k++ {
            if dfs(board, visit, j, k, 0, word) {
                return true
            }
        }
    }
    return false
}

func dfs(board [][]byte, visit [][]bool, i int, j int, start int, word string) bool {
    if start >= len(word) {
        return true
    } else if i <0 || i >= len(board) || j < 0 || j >= len(board[0]) || visit[i][j] == true || board[i][j] != word[start] {
        return false
    }

    visit[i][j] = true
    res := dfs(board, visit, i+1, j, start+1, word) || dfs(board, visit, i-1, j, start+1, word) || dfs(board, visit, i, j+1, start+1, word) || dfs(board, visit, i, j-1, start+1, word)
    
    visit[i][j] = false
    return res
}