地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?
dfs
额外用一个二维数组记录当前点是否已经走过
递归的定义为 当前点是否可以到达 若可以到达 在向右和向下可到达的格子数基础上加1
优化:可以不用递归四个方向 根据k与可到达格子之间的变化规律 可只通过向右和向下两个方向递归遍历所有可到达格子
class Solution {
boolean[][] isVisited;
public int movingCount(int m, int n, int k) {
isVisited = new boolean[m][n];
return dfs(m, n, k, 0, 0);
}
private int dfs(int m, int n, int k, int i, int j){
if(i >= m || j >= n || isVisited[i][j])
return 0;
if(getSum(i) + getSum(j) > k){
isVisited[i][j] = true;
return 0;
}
isVisited[i][j] = true;
return dfs(m, n, k, i + 1, j) + dfs(m, n, k, i, j + 1) + 1;
}
private int getSum(int x){
int res = 0;
while(x > 0){
res += x % 10;
x /= 10;
}
return res;
}
}