剑指offer 第二版（Python3）--面试题13：机器人的运动范围

第2章 面试需要的基础知识

  面试题4：二维数组中的查找

  面试题5：替换空格

  面试题6：从尾到头打印链表

  面试题7：重建二叉树

  面试题9：用两个栈实现队列

  面试题10：斐波那契数列

  面试题11：旋转数组的最小值

  面试题12：矩阵中的路径

  面试题13：机器人的运动范围

  面试题14：剪绳子

  面试题15：二进制数中1的个数

第3章 高质量的代码

第4章 解决面试题的思路

第5章 优化时间和空间效率

第6章 面试中的各项能力

第7章 两个面试案例

---

题目描述
牛客网
  地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

解题思路
  标准的广度优先搜索，当然也可以用深搜。要注意的是已访问的点要记录，否则会重复访问。

实战

class Solution:
    def movingCount(self, threshold, rows, cols):
        # write code here
        # 计算数位之和
        def sums(number):
            counts = 0
            while number % 10:
                n = number % 10
                counts += n
                number = number // 10
            return counts
        
        if rows <= 0 or cols <=0 or threshold < 0:
            return 0
        points = [(0, 0)]
        direction = [(-1, 0), (0, 1), (1, 0), (0, -1)]
        visited = [[0 for _ in range(cols)] for __ in range(rows)]  # 记录已访问的点
        visited[0][0] = 1
        counts = 0  # 记录访问的格子数
        while points:
            point = points.pop()
            counts += 1
            i, j = point[0], point[1]
            for d in direction:
                x, y = i + d[0], j + d[1]
                if x < 0 or y < 0 or x >= rows or y >= cols or visited[x][y] or (sums(x) + sums(y)>threshold): 
                    continue
                points.append((x, y))
                visited[x][y] = 1
        return counts