【Golang】LeetCode-剑指Offer-面试题13-机器人的运动范围

题目

地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。
一个机器人从坐标 [0, 0] 的格子开始移动，它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。

例如，当k为18时，机器人能够进入方格 [35, 37] ，因为3+5+3+7=18。 但它不能进入方格 [35,
38]，因为3+5+3+8=19。请问该机器人能够到达多少个格子？

示例 1：

输入：m = 2, n = 3, k = 1 输出：3

示例 2：

输入：m = 3, n = 1, k = 0 输出：1

提示：

1 <= n,m <= 100
0 <= k <= 20

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof

---

解题思路

解题思路引用自作者：z1m

如何计算行坐标和列坐标的数位之和

• 对于一个整数 n，我们有以下代码模板

func sum (n int)int{
    temp:=0
    for n>0{
        temp+=n%10//获取位数上的数值
        n/=10 	//进位
    }
    return temp
}

• 因此我们只需将行坐标和列坐标分别求数位之和，将其结果相加即可。

哪些格子在运动范围之内

• 机器人的运动范围需要同时满足以下两个条件：

  • 当前格子坐标数位之和不大于 k；
  • 从坐标 (0, 0) 通过上下左右移动可以到达当前格子
    

• 我们可以将矩阵看作一个有向图，而遍历图的方式一般分为

  • 广度优先搜索（BFS）和深度优先搜索（DFS）
  • 因此从 (0, 0) 点出发，采用 BFS 或 DFS
  • 如果当前格子满足以上两个条件，则将结果 ret 加 1。

• 在广度优先算法，由于可行解的连通性和结构，仅考虑向下和向右的移动方向即可，深度优先算法同样如此。

  • 由于题目要求从 (0, 0) 点出发，因此任何一个点都可以只通过向右和向下两个动作达到，因此代码中可以只考虑这两个方向。

---

深度优先搜索 DFS——代码

• 深度优先搜索一般使用栈来实现。
• 本题使用递归可以更轻松实现，我们定义一个递归函数 dfs( )，如果坐标不满足条件，结束递归状态，否则将下一步满足条件的坐标代入递归函数。

–执行用时：0 ms --内存消耗：2 MB

func movingCount(m int, n int, k int) int {
    if k == 0 {
        return 1
    }
    var ret int
    //arr:=[100][100]bool{}
    arr:=make([][]bool,m)
    for i,_:=range arr{
        arr[i]=make([]bool,n)
    }
    var dfs func(row,col int)int
    dfs=func(row,col int)int{
        if row >= m || col >= n || sum(row,col) > k || arr[row][col] {
            return ret
        }
        ret++
        arr[row][col]=true
        return dfs(row+1,col)+dfs(row,col+1)
    }
    dfs(0,0)
    return ret
}

func sum (m,n int)int{
    temp:=0
    for m>0{
        temp+=m%10
        m/=10
    }
    for n>0{
        temp+=n%10
        n/=10
    }
    return temp
}