题目描述:
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,
但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),
因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:
递归判断。
虽然思想总体一样,但方法1提现了回溯(更好),必须要会!
方法2(我的)没有回溯且多了全局变量。
public class Solution {
public static void main(String[] args) {
Solution s = new Solution();
int res = s.movingCount(5, 3, 3);
System.out.println(res);
}
public int movingCount(int threshold, int rows, int cols) {
int[][] arr = new int[rows][cols];//记录是否走过同时记录个数
return move(threshold, rows, cols, 0, 0, arr);
}
public int move(int threadhold, int rows, int cols, int x, int y, int[][] arr) {
//递归结束条件,一旦走到重复的位置就结束,因为回到相同的位置后续步骤已经走过,且会陷入死循环
if (x < 0 || x >= rows || y < 0 || y >= cols
|| (calDigitSum(x) + calDigitSum(y)) > threadhold || arr[x][y] != 0) {
return 0;
}
arr[x][y] = 1;//将当前的位置记为已走过
//!!!!!当前位置和四周可以走的步数,合起来就是总的可以走的步数,回溯的过程
return move(threadhold, rows, cols, x - 1, y, arr) +
move(threadhold, rows, cols, x + 1, y, arr) +
move(threadhold, rows, cols, x, y - 1, arr) +
move(threadhold, rows, cols, x, y + 1, arr) + 1;
}
int count = 0;//记第count个可以走的位置
public int movingCount2(int threshold, int rows, int cols) {
int[][] arr = new int[rows][cols];//记录是否走过同时记录个数
move2(threshold, rows, cols, 0, 0, arr);
return count;
}
public void move2(int threadhold, int rows, int cols, int x, int y, int[][] arr) {
//递归结束条件,一旦走到重复的位置就结束,因为回到相同的位置后续步骤已经走过,且会陷入死循环
if (x < 0 || x >= rows || y < 0 || y >= cols
|| (calDigitSum(x) + calDigitSum(y)) > threadhold || arr[x][y] != 0) {
return;
}
arr[x][y] = ++count;//将当前的位置记为第count个可以走的位置
move2(threadhold, rows, cols, x - 1, y, arr);
move2(threadhold, rows, cols, x + 1, y, arr);
move2(threadhold, rows, cols, x, y - 1, arr);
move2(threadhold, rows, cols, x, y + 1, arr);
}
//计算num的位数和
public int calDigitSum(int num) {
int res = 0;
do {
res += num % 10;
} while ((num = num / 10) != 0);
return res;
}
}