题解 | #机器人的运动范围#

# -*- coding:utf-8 -*-


def canGet(matrix,i,j,rows,cols,thre):
    
    res=int(i/10)+i%10+int(j/10)+j%10
    if i<0&nbs***bsp;i>=rows&nbs***bsp;j<0&nbs***bsp;j>=cols:
        return False
    elif res>thre:
        return False
    if matrix[i*cols+j]:
        return False
    
    
    return True

def dfs(matrix,i,j,rows,cols,dirs,thre):
    cnt=0
    
    if(canGet(matrix,i,j,rows,cols,thre)):
        matrix[i*cols+j]=True
        cnt=1+dfs(matrix, i-1, j, rows, cols, dirs, thre)+dfs(matrix, i+1, j, rows, cols, dirs, thre)+dfs(matrix, i, j-1, rows, cols, dirs, thre)+dfs(matrix, i, j+1, rows, cols, dirs, thre)
        
    
    return cnt
    
    

class Solution:
    def movingCount(self, threshold, rows, cols):
        
        if threshold<0:
            return 0
        if threshold==0:
            return 1
        
        
        dirs=[[0,-1],[0,1],[-1,0],[1,0]]
#         cnt=0
        matrixh=[False]*(rows*cols)
#         for x in range(rows):
#             for y in range(cols):
#                 matrixh.append(0.0)
#         for m in range(rows):
#             for n in range(cols):
#                 matrix[m][n]=0
        i,j=0,0
        cnt=dfs(matrixh, i, j, rows, cols, dirs,threshold)
        return cnt

        
        
        
        # write code here