题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
例子:threshold:10 rows:1 cols:100 结果为29,只能走到(0,28),不会再往下走,所以不能通过遍历每个方格进行求解
public class Solution {
public int movingCount(int threshold, int rows, int cols)
{
if(threshold<0 || rows<1 || cols<1) return 0;
boolean[][] visited = new boolean[rows][cols];
return movingCount(threshold,rows,cols,0,0,visited);
}
public int movingCount(int threshold,int rows,int cols,int i,int j,boolean[][] visited){
if(i<0 || i>=rows || j<0 || j>=cols || visited[i][j]) return 0;
visited[i][j] = true;
if(getDigitSum(i)+getDigitSum(j)<=threshold){
return 1 + movingCount(threshold,rows,cols,i-1,j,visited)
+ movingCount(threshold,rows,cols,i+1,j,visited)
+ movingCount(threshold,rows,cols,i,j-1,visited)
+ movingCount(threshold,rows,cols,i,j+1,visited);
}
return 0;
}
public int getDigitSum(int num){
int sum = 0;
while(num!=0){
sum += num % 10;
num /= 10;
}
return sum;
}
}