题目叙述
地上有一个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 sum_number(number):
sum = 0
for i in str(number):
sum += int(i)
return sum
def dfs(lis,res,i,j,threshold,rows,cols):
if (i < 0 or i > rows - 1): return res
if (j < 0 or j > cols - 1): return res
if(sum_number(i)+sum_number(j)>threshold):
return res
# if(i==rows-1 and j==cols-1): return res
if(lis[i][j]==1): return res
res+=1
lis[i][j]=1
res = dfs(lis,res,i-1,j,threshold,rows,cols) # 向上的
res = dfs(lis,res,i+1,j,threshold,rows,cols) # 向下的
res = dfs(lis,res,i,j-1,threshold,rows,cols) # 向左的
res = dfs(lis,res,i,j+1,threshold,rows,cols) # 向右的
return res
lis = [[0 for i in range(cols)] for i in range(rows)]
res = 0
return dfs(lis, res, 0, 0, threshold, rows, cols)