机器人的运动范围

题目要求

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

算法分析

还是dfs问题，首先生成地图m*n，然后遍历，已遍历过则标记，注意边界值也要判断行坐标和列坐标不能超过k。

代码

class Solution(object):
    def movingCount(self, m, n, k):
        matrix = [[1 for i in range(n)] for j in range(m)] #生成全1的图
        return self.dfs(matrix, 0, 0, k)
    def judge(self, x, y, k):
        res = 0
        while x > 0:
            res += x % 10
            x /= 10
        while y > 0:
            res += y % 10
            y /= 10
        if res > k:
            return False
        else:
            return True
    def dfs(self, matrix, row, col, k):
        if row >= len(matrix) or row < 0 or col >= len(matrix[0]) or col < 0 or matrix[row][col] == 0 or self.judge(row, col, k) == False:
            return 0
        sum = 1 #路径+1
        matrix[row][col] = 0#已遍历
        sum += self.dfs(matrix, row + 1, col, k) + self.dfs(matrix, row - 1, col, k) + self.dfs(matrix, row, col + 1, k) + self.dfs(matrix, row, col - 1, k)
        return sum