剑指offer：机器人的运动范围

题目描述

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

思路：

和https://blog.youkuaiyun.com/orangefly0214/article/details/87935560矩阵中的路径解题思路相近。

可以将这个方格看成m*n的矩阵，在这个矩阵中，除了边界的格子外，其他格子都有4个相邻的格子。

• 使用一个访问数组记录是否已经经过该格子。
• 机器人从(0,0)开始移动，当它准备进入(i,j)的格子时，通过检查坐标的数位来判断机器人是否能够进入。
• 如果机器人能进入(i,j)的格子，接着在判断它是否能进入四个相邻的格子(i,j-1),(i,j+1),(i-1,j),(i+1,j)。

实现：

public class Solution {
    public int movingCount(int threshold, int rows, int cols)
    {
        if(threshold<0||rows<0||cols<0){
            return 0;
        }
        boolean[] flag=new boolean[rows*cols];
        int count=movingCountCore(threshold,rows,cols,0,0,flag);
        return count;
    }
    private int movingCountCore(int threshold,int rows,int cols,int i,int j,boolean[] flag){
        int count=0;
        if(check(threshold,rows,cols,i,j,flag)){
            flag[i*cols+j]=true;
            count=1+movingCountCore(threshold,rows,cols,i-1,j,flag)+
                movingCountCore(threshold,rows,cols,i+1,j,flag)+
                movingCountCore(threshold,rows,cols,i,j-1,flag)+
                movingCountCore(threshold,rows,cols,i,j+1,flag);
        }
        return count;
    }
    private boolean check(int threshold,int rows,int cols,int i,int j,boolean[] flag){
        if(i>=0&&i<rows&&j>=0&&j<cols&&getSum(i)+getSum(j)<=threshold&&!flag[i*cols+j]){
            return true;
        }
        return false;
    }
    private int getSum(int num){
        int sum=0;
        while(num>0){
            sum+=num%10;
            num=num/10;
        }
        return sum;
    }
}

 

测试代码：

public static void main(String[] args) {
		movingCountDemo mcd=new movingCountDemo();
		Scanner scanner = new Scanner(System.in);
		System.out.println("请输入限制条件k:");
		int k = scanner.nextInt();
		System.out.println("请输入方格的行数m：");
		int m = scanner.nextInt();
		System.out.println("请输入方格的列数n:");
		int n = scanner.nextInt();
		System.out.println("矩阵能到达的方格数是：");
		System.out.println(mcd.movingCount(k, m, n));

	}

结果：

 

 

参考：

https://blog.youkuaiyun.com/gg543012991/article/details/52900234

https://www.cnblogs.com/rosending/p/5742725.html