题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
tips: bfs,记得设置isVisited矩阵,否则可能会复杂度过大。
书上也有用递归做的,代码比较简洁。
class Solution {
public:
bool isVaildPos(int threshold, 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 <= threshold;
}
int movingCount(int threshold, int rows, int cols)
{
if(rows<=0||cols<=0||threshold<0) {
return 0;
}
vector<vector<bool>> isVisited=vector<vector<bool>>(rows,vector<bool>(cols,false));
queue<pair<int,int>> que;
if (isVaildPos(threshold,0,0))
{
isVisited[0][0]=true;
que.push({0,0});
}
vector<pair<int,int>> move={{-1,0},{0,1},{1,0},{0,-1}};
int count=0;
while(!que.empty()) {
pair<int,int> pos=que.front();
que.pop();
count++;
for (int i = 0; i < move.size(); i++)
{
int nextPosX=pos.first+move[i].first;
int nextPosY=pos.second+move[i].second;
if(nextPosX<0||nextPosX>=rows||nextPosY<0||nextPosY>=cols) {
continue;
}
if(isVisited[nextPosX][nextPosY]==false&&isVaildPos(threshold,nextPosX,nextPosY)) {
isVisited[nextPosX][nextPosY]=true;
que.push({nextPosX,nextPosY});
}
}
}
return count;
}
};