第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