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

题目描述

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

思路

设置标记变量：为1表示不可达，为0表示可达
递归
递归出口：越界，threshold值非法，数位和大于threshold，格子不可达
寻找可达路径，则整个路径是连通的，所以不用考虑回溯

实现

public class Solution {
    public int movingCount(int threshold, int rows, int cols)
    {
        int[][] flag=new int[rows][cols];    //标记每格是否可达：初始都是1（不可达），若可达的话置0
        return moving(threshold,rows,cols,flag,0,0);  //从（0,0）开始
    }
    
    public int moving(int threshold,int rows,int cols,int[][] flag,int i,int j){
        //递归出口（越界，threshold值非法，数位和大于threshold，格子不可达）
        if(i<0 || j<0 || i>=rows || j>=cols || threshold<0 || (getSum(i)+getSum(j)>threshold) || flag[i][j]==1){
            return 0;
        }
        flag[i][j]=1;  //该方格无法到达
        
        //没有以上情况，就朝四个方向试探是否可达，最终可达格子数是四个方向之和+1（当前位置i,j）（整个路径是连通的，所以不用考虑回溯）
        return moving(threshold,rows,cols,flag,i+1,j)+
               moving(threshold,rows,cols,flag,i-1,j)+
               moving(threshold,rows,cols,flag,i,j+1)+
               moving(threshold,rows,cols,flag,i,j-1)+1;
    }
    
    //求数位和
    public int getSum(int i){
        int sum=0;
        while(i!=0){
            sum+=i%10;
            i=i/10;
        }
        return sum;
    }
    
}