题目:地上有一个m行n列d方格。一个机器人从坐标(0,0)的格子开始移动,它每次可以向左、右、上、下移动一格,但不能进入行坐标和列坐标的数位之和大于k的格子。
def robot_move(rows,cols,k):
if rows<0 or cols<0 or k<0:
return 0
visited = [[0 for i in range(cols)] for j in range(rows)]
return move(rows,cols,0,0,visited,k)
def move(rows,cols,row,col,visited,k):
cnt=0
if check(rows,cols,row,col,k,visited):
visited[row][col]=1
cnt = 1+move(rows,cols,row-1,col,visited,k)+move(rows,cols,row+1,col,visited,k)+\
+move(rows,cols,row,col-1,visited,k)+move(rows,cols,row,col+1,visited,k)
return cnt
def check(rows,cols,row,col,k,visited):
# print('check:', row, col,get_nums_sum(row),get_nums_sum(col),visited[row][col])
if row >=0 and col>=0 and col<cols and row<rows and get_nums_sum(row)+get_nums_sum(col)<=k and visited[row][col]==0:
return True
return False
def get_nums_sum(nums):
if nums<0:
return 0
res=0
while nums:
res+=nums%10
nums = nums//10
return res
注:
visited:存储该位置是否已经走过
check():验证该格子是否可以走
get_nums_sum():计算该格子坐标和
move():如果格子可以走,往四个方向递归查找