[剑指 Offer] 13. 机器人的运动范围

地上有一个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。请问该机器人能够到达多少个格子？

 

示例 1：

输入：m = 2, n = 3, k = 1
输出：3

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
 

使用广度优先算法求解，用mat[m][n]矩阵来标记(i,j)是否被访问，0表示未被访问，1表示已经被访问。

#python
class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        def valid(a,b):
            tmp1 = 0
            tmp2 = 0
            while a:
                tmp1+=a%10
                a = a//10
            while b:
                tmp2+=b%10
                b = b//10
            return (tmp1+tmp2)<=k
        mat = [[0]*n for _ in range(m)]
        que = deque()
        ans = 0
        if valid(0,0):
            que.append((0,0))
            ans+=1
            mat[0][0]=1
        while que:
            i,j = que.popleft()
            for x,y in ((1,0),(0,1),(-1,0),(0,-1)):
                nx,ny = i+x,j+y
                if 0<=nx<m and 0<=ny<n and mat[nx][ny]==0 and valid(nx,ny):
                    que.append((nx,ny))
                    ans+=1
                    mat[nx][ny]=1
        return ans