题目概述
- 算法说明
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
范围:
1 <= rows, cols<= 100
0 <= threshold <= 20 - 测试用例
案例1:
输入:5,10,10 输出:21
案例2:
输入:15,20,20 输出:359
解析&参考答案
- 解析
使用递归的方式获取最终结果,起始地点为(0,0),在递归中返回 1+递归当前点右边点+递归当前点下面点,递归的时候需要使用一个map记录当前点是否已经走过(走过则返回0). - 参考答案
vim jz66.go
package main
import "fmt"
func GetPositionSum(row, col int) int {
sum := 0
for row != 0 {
sum += row % 10
row = row / 10
}
for col != 0 {
sum += col % 10
col = col / 10
}
return sum
}
func MovingCountCore(threshold, r, c, rows, cols int, m map[int]int) int {
if r >= rows || c >= cols { // 此处需要>=,否则会扩大范围
return 0
}
if m[r*cols+c] == 1 || GetPositionSum(r, c) > threshold {
return 0
}
result := 0
m[r*cols+c] = 1
result = 1 + MovingCountCore(threshold, r+1, c, rows, cols, m) + MovingCountCore(threshold, r, c+1, rows, cols, m)
return result
}
func MovingCount(threshold int, rows int, cols int) int {
if rows == 0 || cols == 0 {
return 0
}
m := make(map[int]int)
result := MovingCountCore(threshold, 0, 0, rows, cols, m)
return result
}
func main() {
// threshold, rows, cols := 5, 10, 10
threshold, rows, cols := 15, 20, 20
result := MovingCount(threshold, rows, cols)
fmt.Println(result)
}
注意事项
- to add
说明
- 当前使用 go1.15.8
- 参考 牛客网--剑指offer
标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目,例如 jz01 代表牛客网剑指offer中01号题目。
注意!!!
- 笔者最近在学习 golang,因此趁机通过数据结构和算法来进一步熟悉下go语言
- 当前算法主要来源于剑指 offer,后续会进一步补充 LeetCode 上重要算法,以及一些经典算法
- 此处答案仅为参考,不一定是最优解,欢迎感兴趣的读者在评论区提供更优解