地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
这道题我的理解是,如果一个格子可以到达的话,那它必须满足:其上下左右的格子,必须至少有一个是能到达的,还有就是当前格子的行列坐标之和得小于等于给定的k。下面是我的代码
class Solution {
public:
int sum(int num){
int sum=0;
while(num){
sum+=(num%10);
num/=10;
}
return sum;
}
bool haspath(int row,int col,int threshold){
if(row<=1 && col<=1 && !(row == 1 && col == 1)) return 1;
if(row >= 1&&col<1){
if(haspath(row-1,col,threshold)){
if(sum(row)+sum(col)<=threshold){
return 1;
}
return 0;
}
}
else if(row < 1&&col>=1){
if(haspath(row,col-1,threshold)){
if(sum(row)+sum(col)<=threshold){
return 1;
}
return 0;
}
}
else if(row >= 1&&col>=1){
if(haspath(row-1,col,threshold)||haspath(row,col-1,threshold)){
if(sum(row)+sum(col)<=threshold){
return 1;
}
return 0;
}
}
return 0;
}
int movingCount(int threshold, int rows, int cols)
{
if(threshold < 0) return 0;
if(threshold == 0) return 1;
int count = 0;
for(int i = 0;i < rows; i++){
for(int j = 0; j < cols; j++){
if(haspath(i,j,threshold)){
count++;
}
}
}
return count;
}
};
这里的sum函数是计算数位之和的,比较简单,不多赘述。
movingCount函数则是遍历所有的点,判断是否有path可以到达这个点,若可以,则令count+1。
haspath函数,因为遍历顺序是从左往右,从上往下的,所以并不需要把上下左右4种情况都考虑进去,只需要考虑上和左两种情况就行。因为要考虑到防止数组越界的问题,所以我分成4种情况
第一种情况:row<=1 && col<=1 && !(row == 1 && col == 1) 就是(0,0)(0,1)(1,0)这三个点,直接返回1;
第二种情况:row >= 1&&col<1 就是左边界的点,此时,只要判断其上方的点是否可以到达,并且sum也满足条件的话,就返回1
第三种情况:row < 1&&col>=1 就是上边界的点,此时,只要判断其左方的点是否可以到达,并且sum也满足条件的话,就返回1
第四种情况:row >= 1&&col>=1 就是其余的点,此时需要判断左方和上方的点是否至少有一个可以到达,并且sum也满足条件,就返回1
就先这样了吧...遛了