题目描述
地上有一个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)
{
if(threshold < 0)return 0;
r = rows;
c = cols;
t = threshold;
memset(visit,0,sizeof(visit));
dfs(0,0);
return sum;
}
void dfs(int x, int y){
if(!reach(x,y))
return;
if(visit[x][y] || x < 0 || x >= r || y < 0 || y >= c)return;
sum++;
visit[x][y] = 1;
dfs(x+1,y);
dfs(x-1,y);
dfs(x,y+1);
dfs(x,y-1);
}
bool reach(int x, int y){
int s = 0;
while(x){
s += x % 10;
x /= 10;
}
while(y){
s += y % 10;
y /= 10;
}
return s <= t;
}
private:
int sum = 0;
int r = 0;
int c = 0;
int t;
int visit[110][110];
};