题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:bfs
从(0,0)开始判断当前格子数否符合要求,然后返回1+四个方向上符合要求的格子数
class Solution {
public:
int movingCount(int threshold, int rows, int cols)
{
//建立访问数组
bool* visited = new bool[rows*cols];
for(int i = 0;i < rows*cols;i++)
visited[i] = false;
//从(0,0)开始计算格子数
int cnt = Count(threshold,rows,cols,0,0,visited);
//记得释放内存空间
delete [] visited;
return cnt;
}
int Count(int threshold, int rows, int cols,int i,int j,bool *visited)
{
//数组越界或已访问过就返回0
if(i >= rows || j >= cols || i < 0 || j < 0 || visited[i * cols + j])
return 0;
//标记为访问过
visited[i * cols + j] = true;
//计算数位之和并判断
if(bitCnt(i) + bitCnt(j) > threshold)
return 0;
//返回上下左右和本身能达到的格子数
return 1 + Count(threshold,rows,cols,i-1,j,visited)
+ Count(threshold,rows,cols,i+1,j,visited)
+ Count(threshold,rows,cols,i,j-1,visited)
+ Count(threshold,rows,cols,i,j+1,visited);
}
int bitCnt(int i){
int cnt = 0;
while(i != 0){
cnt += i%10;
i = i / 10;
}
return cnt;
}
};