数组17--机器人的运动范围

题目概述

• 算法说明
  地上有一个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)
}

注意事项

1. to add

说明

1. 当前使用 go1.15.8
2. 参考 牛客网--剑指offer
   标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

• 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
• 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解