【剑指Offer】66机器人的运动范围

题目描述

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

时间限制：1秒；空间限制：32768K

解题思路

回溯法递归实现求解。难点在于找到回溯条件。

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.visited = [] #用来记录格子是否数过
        
    def movingCount(self, threshold, rows, cols):
        # write code here
        self.visited = [[0 for i in range(cols+1)] for j in range(rows+1)]
        return self.pd(threshold,rows,cols,0,0)
    
    # 判断一个位置上下左右有没有符合条件并且标志位为1的数
    def pd(self,threshold,rows,cols,r,c):
        # 回溯条件
        if self.visited[r][c] == 1 or r<0 or r>=rows or c<0 or c>=cols or self.bitSum(r,c,threshold)==False:
            return 0
        else:
            self.visited[r][c] = 1
        up,down,left,right = 0,0,0,0
        up = self.pd(threshold,rows,cols,r-1,c)
        down = self.pd(threshold,rows,cols,r+1,c)
        left = self.pd(threshold,rows,cols,r,c-1)
        right = self.pd(threshold,rows,cols,r,c+1)
        return up+down+left+right+1
    
    # 判断x，y的数位之和是否大于k
    def bitSum(self, x, y, k):
        count = 0
        while x!=0:
            count += x%10
            x = x//10
        while y!=0:
            count += y%10
            y = y//10
        return (count <= k)