题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:很简单的bfs应用,注意memset初始化在牛客这个判题机上会出问题,不是默认的全部是false。代码如下:
class Solution {
public:
int is_sum(int value)
{
int cnt= 0;
while(value)
{
int mod = value%10;
cnt+=mod;
value /= 10;
}
return cnt;
}
bool isJudge(int rx,int ry,int threshold)
{
if(rx<0||ry<0) return false;
int temp_x = is_sum(rx);
int temp_y = is_sum(ry);
if(temp_x+temp_y>threshold) return false;
return true;
}
bool out_of_bound(int r_x,int r_y,int rows,int cols)
{
if(r_x<0||r_x>rows||r_y<0||r_y>cols) return false;
return true;
}
int moving(int x,int y,int threshold,int rows,int cols,bool vis[])
{
int dx[]={+1,-1,0,0};
int dy[]={0,0,-1,+1};
int sum = 0;
if(isJudge(x,y,threshold)&&!vis[x*cols+y]&&out_of_bound(x,y,rows-1,cols-1))
{
vis[x*cols+y] = true;
for(int i=0;i<4;i++)
{
int r_x = x + dx[i];
int r_y = y + dy[i];
sum+= moving(r_x,r_y,threshold,rows,cols,vis);
}
sum+=1;
}
return sum;
}
int movingCount(int threshold, int rows, int cols)
{
bool *vis = new bool[rows*cols];
for(int i=0;i<rows*cols;i++) vis[i] = false;
int ans = moving(0,0,threshold,rows,cols,vis);
return ans;
}
};