一、题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行
坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方
格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
二、解题思路
(1)这个方格可以看成 mxn的矩阵,除边界外,其他格子都有4个相邻的格子
(2)机器人从坐标(0,0)开始移动
(3)当它准备进入坐标为(i , j)的格子时,通过检查坐标的数位来判断机器人是否能够进入;
(4)如果机器人能够进入坐标为(i,j)的格子,则再判断它能否进入4个相邻的格子
三、代码实现
public class Solution {
public int movingCount(int threshold, int rows, int cols)
{
//特殊情况
if(threshold<0 || rows<=0 || cols<=0)
return 0;
boolean[] visited = new boolean[rows*cols];//标识路径是否经过某个格子
//将visited矩阵中的每一个值初始化为false
for(int i=0; i<rows*cols; i++){
visited[i] = false;
}
int count = movingCountCore(threshold, rows, cols, 0, 0, visited);
return count;
}
//统计移动了多少个格子
public int movingCountCore(int threshold, int rows, int cols, int row, int col, boolean[] visited){
int count = 0;
if(check(threshold, rows, cols, row, col, visited)){
visited[row*cols +col]=true;
count = 1+ movingCountCore(threshold, rows, cols, row-1, col, visited)
+movingCountCore(threshold, rows, cols, row, col-1, visited)
+movingCountCore(threshold, rows, cols, row+1, col, visited)
+movingCountCore(threshold, rows, cols, row, col+1, visited);
}
return count;
}
//判断机器人能否进入坐标为(row, col)的方格
public boolean check(int threshold, int rows, int cols, int row, int col, boolean[] visited){
if(row>=0 && row<rows && col>=0 && col<cols && getDigitSum(row)+getDigitSum(col)<=threshold && !visited[row*cols+col]){
return true;
}
return false;
}
//用来得到一个数字的数位之和
public int getDigitSum(int number){
int sum=0;
while(number>0){
sum+=number % 10;
number /= 10;
}
return sum;
}
}
四、测试
public static void main(String[] args) {
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();
MovingCount count = new MovingCount();
int movingCount = count.movingCount(k, m, n);
System.out.println("机器人走了:" + movingCount + "步");
}