Python3-[12]面试题13.机器人的运动范围（DFS）A

题目

地上有一个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。请问该机器人能够到达多少个格子？
P.S. 题目来源于leetcode,这是我的题解

解题思路

从（0，0）开始，向右&下搜索即可搜索所有满足要求的格子。若向上/左搜索，则会超时=>出现重复循环。

1.数位和

求出num的各数位相加和s，如num=123,s=6.

def dSum(num):#求取num各位数和,如12=>1+2
    s = 0
    while num != 0:
        s += num % 10
        num = num // 10
    return s

2.DFS

1.递归基：

当前坐标i/j越界、数位和不满足、(i,j)已访问=>返回0

if m <= i or n <= j or k < dSum(i)+dSum(j) or (i,j) in visited: return 0 

2.当前点处理

把当前点（i，j）设为已经访问

visited.add((i,j))

3.尾递归

搜索到（i，j）时的结果为向右&下搜索的结果加1（当前(i,j)是符合要求的格子）。

return 1+DFS(i+1,j)+DFS(i,j+1)

P.S.:图解可以参看这篇文章。

代码

class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        #DFS
        #向下或向右移动搜索就足够
        def dSum(num):#求取num各位数和,如12=>1+2
            s = 0
            while num != 0:
                s += num % 10
                num = num // 10
            return s
        def DFS(i,j):#搜索到（i，j）时的结果为右&下搜索的结果加1
            #递归基：越界、数位和不满足、已访问=>返回0
            if m <= i or n <= j or k < dSum(i)+dSum(j) or (i,j) in visited: return 0 
            #当前节点处理=>添加访问节点
            visited.add((i,j))
            return 1+DFS(i+1,j)+DFS(i,j+1)#自身能达到（1）+右搜结果+下搜结果
        visited = set()
        return DFS(0,0)