地上有一个 m 行和 n 列的方格。一个机器人从坐标 (0, 0) 的格子开始移动,每一次只能向左右上下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于 k 的格子。
例如,当 k 为 18 时,机器人能够进入方格 (35,37),因为 3+5+3+7=18。但是,它不能进入方格 (35,38),因为 3+5+3+8=19。请问该机器人能够达到多少个格子?
import java.util.*;
public class Solution {
private static final int[][] next={{0,-1},{0,1},{-1,0},{1,0}};
private int cnt = 0;
private int rows;
private int cols;
private int threshold;
private int[][] digitSum;
public int movingCount(int threshold, int rows, int cols)
{
this.threshold=threshold;
this.rows = rows;
this.cols = cols;
initDigitSum();
boolean[][] marked = new boolean[rows][cols];
dfs(marked, 0, 0);
return cnt;
}
private void dfs(boolean[][] marked, int r, int c){
if(r<0||c<0||r>=rows||c>=cols||marked[r][c]){
return;
}
marked[r][c]=true;
if(this.digitSum[r][c] > this.threshold){
return;
}
cnt++;
for(int[] n:next){
dfs(marked, r+n[0],c+n[1]);
}
}
private void initDigitSum(){
int[] digitSumOne = new int[Math.max(rows,cols)];
for(int i=0;i< digitSumOne.length;i++){
int n=i;
while(n>0){
digitSumOne[i]+=n%10;
n /=10;
}
}
this.digitSum = new int[rows][cols];
for(int i=0;i< this.rows;i++){
for(int j=0;j<this.cols;j++){
this.digitSum[i][j]=digitSumOne[i]+digitSumOne[j];
}
}
}
}
错误:r和c是大于等于rows和cols时报错,有等于号,他们是从0开始的;
递归那里r+n[0],c+n[1]都写成r了。
分成三个部分,第一部分定义参数,定义标记数组;
第二部分进行递归比较;
第三部分计算二维数组里面行数和列数都数位之和,使用一维数组的循环来做,然后再转换成二维数组对应的值。