JZ66 機器人的運動範圍

機器人的運動範圍

地上有一個m行和n列的方格。一個機器人從座標0,0的格子開始移動，每一次只能向左，右，上，下四個方向移動一格，但是不能進入行座標和列座標的數位之和大於k的格子。 例如，當k爲18時，機器人能夠進入方格（35,37），因爲3+5+3+7 = 18。但是，它不能進入方格（35,38），因爲3+5+3+8 = 19。請問該機器人能夠達到多少個格子？

思路：

 

 

 

func movingCount(m int, n int, k int) int {
    dp := make([][]int, m+1)
    for i := range dp {
        dp[i] = make([]int, n+1)
    }

    return dfs(m, n, 0, 0, k, dp)
}

func dfs(m, n, i, j, k int, dp [][]int) int {
    if i < 0 || j < 0 || i >= m || j >= n || dp[i][j] == 1 || (sumPos(i)+sumPos(j)) > k {
        return 0
    }

    dp[i][j] = 1

    sum := 1
    sum += dfs(m, n, i, j+1, k, dp)
    sum += dfs(m, n, i, j-1, k, dp)
    sum += dfs(m, n, i+1, j, k, dp)
    sum += dfs(m, n, i-1, j, k, dp)
    return sum
}

// 求所有位之和
func sumPos(n int) int {
    var sum int

    for n > 0 {
        sum += n % 10
        n = n / 10
    }

    return sum
}