题目描述:
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
本题知识点:
数组
解题思路:
根据方格大小设置一个标志位数组,如果该点已经计算过了,就将该点的值设为true,从(0,0)开始每次对上下左右四个点进行查找,如果该点的坐标没有越界,该点没有被统计过,并且满足题目的阈值条件,则number+1,否则不进行处理。
代码:
class Solution {
public:
int movingCount(int threshold, int rows, int cols)
{
if(threshold < 0 || rows == 0 || cols == 0)
return 0;
bool *passed = new bool[rows * cols]();
int x = 0, y = 0, number= 0;
func(threshold, rows, cols, x, y, number, passed);
return number;
}
int Calsum(int x, int y)
{
int sum = 0;
while(x != 0)
{
sum += x % 10;
x /= 10;
}
while(y != 0)
{
sum += y % 10;
y /= 10;
}
return sum;
}
void func(int threshold, int rows, int cols, int x, int y, int& number, bool *passed)
{
if(x >= rows || y >= cols || x < 0 || y < 0 || passed[x * cols + y] || Calsum(x, y) > threshold)
return;
passed[x * cols + y] = true;
number++;
func(threshold, rows, cols, x + 1, y, number, passed);
func(threshold, rows, cols, x - 1, y, number, passed);
func(threshold, rows, cols, x, y + 1, number, passed);
func(threshold, rows, cols, x, y - 1, number, passed);
}
};