题目描述
地上有一个m行和n列的方格。一个机器人从座标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行座标和列座标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:
1.使用二维数组记录方格信息,true:机器人能走到;false:机器人走不到
2.从座标[0,0]开始往下走,如果当前座标符合要求,数组置为true,如果数组已经为true,说明已经走过,为了防止循环递归,数组为true时,返回。
3.递归调用附近四个座标
代码:
public class Solution { public static int movingCount(int threshold, int rows, int cols) { boolean[][] path = new boolean[rows][cols]; movingCount(path, rows, cols, 0, 0, threshold); int num = 0; for (boolean[] p : path) { for (boolean b : p) { if (b) { num++; } } } return num; } public static void movingCount(boolean[][] path, int rows, int cols, int x, int y, int result) { if (checkPath(path, rows, cols, x, y, result)) { path[x][y] = true; movingCount(path, rows, cols, x - 1, y, result); movingCount(path, rows, cols, x + 1, y, result); movingCount(path, rows, cols, x, y - 1, result); movingCount(path, rows, cols, x, y + 1, result); } } public static boolean checkPath(boolean[][] path, int rows, int cols, int x, int y, int result) { if (x < 0 || x >= rows || y < 0 || y >= cols) { return false; } if (path[x][y]) { return false; } int k = 0; while (x != 0) { k += x % 10; x = x / 10; } while (y != 0) { k += y % 10; y = y / 10; } return k <= result; } }