剑指offer 13 机器人运动范围

地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        def judge(m,n):
            return sum(list(map(int,str(m)+str(n))))
        alist=set([(0,0)])
        for i in range(m):
            for j in range(n):
                if ((i-1,j) in alist or (i,j-1) in alist) and judge(i,j)<=k:
                    alist.add((i,j))
        return len(alist)


该方法可能超时
class Solution:
    def movingCount(self, threshold, rows, cols):  # 主函数，threshold是k值。
        # write code here rows行，cols列
        visited = [[False] * cols for i in range(rows)]  # 把visit全部置false
        def get_sum(x, y):  # 行坐标和列坐标的数位之和
            return sum(list(map(int, str(x) + str(y))))

        def movingcode(threshold, rows, cols, i, j):  # 功能：如果小于k值，就置为True。接着找四周
            if get_sum(i, j) <= threshold:  # 如果<=k,就执行；否则没有动作
                visited[i][j] = True
                for x, y in ((i - 1, j), (i + 1, j), (i, j - 1), (i, j + 1)):  # 上下左右遍历
                    if 0 <= x < rows and 0 <= y < cols and visited[x][y] == False:  # 如果当前位置是False
                        movingcode(threshold, rows, cols, x, y)

        movingcode(threshold, rows, cols, 0, 0)
        print(visited)
        p=sum(visited, [])
        print(p)
        return sum(sum(visited, []))  # 降维
c=Solution()
t=c.movingCount(4,4,5)
print(t)