13-机器人的运动范围-python

题目：地上有一个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():如果格子可以走，往四个方向递归查找