题目:
地上有一个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
思路:
因为已经做过剑指offer12题,所以容易想到用递归函数来做。同样使用深度优先搜索方法,通过分析机器人只能从右或下两个方向进行行走才能到达满足条件的格子(可达解)。有大佬用数位和增量方法进行求解,但是我做题时根本想不到,感觉不被容易想到,所以没有写在文章里。
递归终止条件为:坐标不满足条件
代码:
class Solution:
def movingCount(self, m: int, n: int, k: int) -> int:
def sumofDigit(x, y):
result = 0
while x > 0:
result += x % 10
x //= 10
while y > 0:
result += y % 10
y //= 10
return result
def dfs(i, j):
if i == m or j == n or sumofDigit(i, j) > k or (i, j) in marked:
return
marked.add((i, j))
dfs(i + 1, j)
dfs(i, j + 1)
marked = set()
dfs(0, 0)
return len(marked)
复杂度:
- 时间复杂度:O(m*n)
- 空间复杂度:O(m*n)