剑指offer--面试题67：机器人的运动范围

题目描述

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

python实现：

# -*- coding:utf-8 -*-
class Solution:
    result = 0
    def movingCount(self, threshold, rows, cols):
        # write code here
        if rows<=0 or cols<=0:
            return 0
        visited = [ [False]*cols for _ in range(rows) ]
         
        def getDigitSum(num):
            sum = 0
            while num>0:
                num, tmpSum = divmod(num, 10)
                sum += tmpSum
            return sum
         
        def isValid(i, j):
            if not(0<=i<rows and 0<=j<cols):
                return False
            if getDigitSum(i)+getDigitSum(j)>threshold:
                return False
            #print i, j
            return visited[i][j] != True
             
        def dfs(i, j):
            if isValid(i, j):
                visited[i][j] = True
                self.result += 1
                dfs(i-1, j)
                dfs(i, j+1)
                dfs(i+1, j)
                dfs(i, j-1)
         
        dfs(0,0)
        return self.result