剑指Offer:机器人的运动范围Java/Python

1.题目描述

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

2.算法描述

方法：BFS

$1.初始化一个与方格同等大小的数组flag，用来标识机器人是否可以活动到此，初始时flag为全0。$
$2.机器人从方格左上角，i,j为0,位置开始递归调用$
$如果i，j位置不合理，或这个位置已经探索过，或这个位置不满足threshold要求，则返回0。$
$否则，把这个位置的flag标识为1，表示这个位置探索过和机器人可以到达，以后不能重复探索。$
$以这个位置为起始点，递归地探索上、下、左、右四个方向的可运动范围，返回四个运动范围的数量+1。$

3.代码描述

3.1.Java代码

public class Solution {
    public int movingCount(int threshold, int rows, int cols)
    {
        int[][] flag = new int[rows][cols];//用来标识位置机器人是否可以到此位置
        return helper(0, 0, rows, cols, flag, threshold);
    }
    private int helper(int i, int j, int rows, int cols, int[][] flag, int threshold){
        //位置不合理  位置已经被探索  位置不满足要求 返回0
        if(i<0||i>=rows||j<0||j>=cols||flag[i][j]==1||sum(i,j)>threshold) return 0;
        //位置合理 满足要求
        flag[i][j] = 1;
        //返回1 + 从这个位置上、下、左、右探索的可行位置
        return helper(i+1, j, rows, cols, flag, threshold)+
               helper(i-1, j, rows, cols, flag, threshold)+
               helper(i, j+1, rows, cols, flag, threshold)+
               helper(i, j-1, rows, cols, flag, threshold)+1;
    }
    private int sum(int i, int j){
        int ans = 0;
        while(i!=0){
            ans += i % 10;
            i /= 10;
        }
        while(j!=0){
            ans += j % 10;
            j /= 10;
        }
        return ans;
    }
}

3.2.Python代码

# -*- coding:utf-8 -*-
class Solution:
    def movingCount(self, threshold, rows, cols):
        flag = [[0 for j in range(cols)] for i in range(rows)]
        return self.helper(0, 0, rows, cols, flag, threshold)
    def helper(self, i, j, rows, cols, flag, threshold):
        if i<0 or i>=rows or j<0 or j>=cols or flag[i][j]==1 or self.fun(i,j)>threshold:
            return 0
        flag[i][j] = 1
        return  (self.helper(i-1, j, rows, cols, flag, threshold) + 
                self.helper(i+1, j, rows, cols, flag, threshold) + 
                self.helper(i, j-1, rows, cols, flag, threshold) + 
                self.helper(i, j+1, rows, cols, flag, threshold)+1)
    def fun(self, i, j):
        ans = 0
        while i:
            ans += i % 10
            i //= 10
        while j:
            ans += j % 10
            j //= 10
        return ans