题目描述
地上有一个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