机器人的运动范围
题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:这一题看上去其实和之前的矩阵中的路径相似,可以把他转换成走迷宫,起点是(0,0)终点是没一个值,限制条件就是每格的位数和不大于k;但这样显然复杂度高。这题不同之处是只要求能走到多少格,没有所示一条路走到底,所以这就是差异点。
public class 机器人的运动范围 {
public static class Solution {
public int movingCount(int threshold, int rows, int cols) {
int map[][] = new int[rows][cols];
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
map[i][j] = 0;
}
}
return movingCount2(threshold, rows, cols, map, 0, 0);
}
public int movingCount2(int threshold, int rows, int cols, int[][] map, int x, int y) {
int count=0;
if(x >= 0 && x < rows && y >= 0 && y < cols){
if (map[x][y] == 0 && check(x, y, threshold)){
map[x][y]=1;
count=1;
int active[][] = new int[][]{{0, -1}, {0, 1}, {1, 0}, {-1, 0}};
for (int i = 0; i < active.length; i++) {
int x2 = x + active[i][0];
int y2 = y + active[i][1];
count = count +movingCount2(threshold, rows, cols, map, x2, y2);
}
}
}
return count;
}
public boolean check(int x, int y, int threshold) {
int sum = 0;
while (x > 0) {
sum = sum + x % 10;
x = x / 10;
}
while (y > 0) {
sum = sum + y % 10;
y = y / 10;
}
if (sum <= threshold) {
return true;
} else {
return false;
}
}
public static void main(String[] args) {
Solution solution = new Solution();
int n = solution.movingCount(5, 10, 10);
System.out.println(n);
}
}
}