题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
分析
这还是回朔的思想。
我们可以从第一个格子(0, 0)开始出发,让机器人上、下、左、右进行移动,并判断每次在当前格子上是否满足条件,为了记录当前格子是否走过,还需要定义一个int标志数组。对于第n步,如果上下左右都不满足条件,那么需要退到n-1步继续执行,可以利用递归的思想。
代码
public class Solution {
public int movingCount(int threshold, int rows, int cols) {
int[] flag = new int[rows * cols];
return helper(threshold, rows, cols, 0, 0, flag);
}
public int helper(int k, int rows, int cols, int i, int j, int[] flag) {
if(i < 0 || i >= rows || j < 0 || j >= cols || (getDigit(i) + getDigit(j) > k) || flag[i * cols + j] == 1) {
return 0;
}
flag[i * cols + j] = 1;
return helper(k, rows, cols, i - 1, j, flag)
+ helper(k, rows, cols, i + 1, j, flag)
+ helper(k, rows, cols, i, j - 1, flag)
+ helper(k, rows, cols, i, j + 1, flag)
+ 1;
}
public int getDigit(int num) {
int digit = Integer.toString(num).length();
int nums = 0;
for (int i = 1, j = 0; j < digit; i *= 10, j++) {
nums += ((num / i) % 10);
}
return nums;
}
}