题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
class Solution {
public:
int movingCount(int threshold, int rows, int cols)
{
bool graa[rows][cols];
memset(graa,0,sizeof graa);
int count=0 ;
function<void(int,int)> fun ;
bool *gra= (bool*)graa ;
fun = [&](int x,int y){
if(*(gra+x*rows+y)==false){
*(gra+x*rows+y)= true ;
if(num(x)+num(y)<=threshold){
count++ ;
if(x-1>=0) fun(x-1,y);
if(x+1<cols) fun(x+1,y);
if(y-1>=0) fun(x,y-1);
if(y+1<rows) fun(x,y+1);
}
}
};
fun(0,0);
return count ;
}
int num(int p){
int sum = 0 ;
while(p!=0){
sum += p%10 ;
p /=10 ;
}
return sum ;
}
};