【题目描述】
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18(注意个位+十位)。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
【解法】
1.dfs
class Solution {
public:
using V = vector<int>;
using VV = vector<V>;
int dir[5] = {-1, 0, 1, 0, -1};
int check(int n) {
int sum = 0;
while (n) {
sum += (n % 10);
n /= 10;
}
return sum;
}
void dfs(int x, int y, int sho, int r, int c, int &ret, VV &mark) {
// 检查下标 和 是否访问
if (x < 0 || x >= r || y < 0 || y >= c || mark[x][y] == 1) {
return;
}
// 检查当前坐标是否满足条件
if (check(x) + check(y) > sho) {
return;
}
// 代码走到这里,说明当前坐标符合条件
mark[x][y] = 1;
ret += 1;
for (int i = 0; i < 4; ++i) {
dfs(x + dir[i], y + dir[i + 1], sho, r, c, ret, mark);
}
}
int movingCount(int sho, int rows, int cols)
{
if (sho <= 0) {
return 0;
}
VV mark(rows, V(cols, -1));//mark是个矩阵
int ret = 0;
dfs(0, 0, sho, rows, cols, ret, mark);
return ret;
}
};
2.bfs
class Solution {
public:
using pii = pair<int,int>;
int dir[5] = {-1, 0, 1, 0, -1};
int check(int n) {
int sum = 0;
while (n) {
sum += (n % 10);
n /= 10;
}
return sum;
}
int movingCount(int sho, int rows, int cols)
{
if (sho <= 0) {
return 0;
}
int ret = 0;
int mark[rows][cols];
memset(mark, -1, sizeof(mark));
queue<pii> q;
q.push({0, 0});
mark[0][0] = 1;
while (!q.empty()) {
auto node = q.front();
q.pop();
// 每次保证进队列的都是满足条件的坐标
++ret;
for (int i = 0; i < 4; ++i) {
int x = node.first + dir[i];
int y = node.second + dir[i + 1];
if (x >= 0 && x < rows && y >= 0 && y < cols && mark[x][y] == -1) {
if (check(x) + check(y) <= sho) {
q.push({x, y});
mark[x][y] = 1;
}
}
}
}
return ret;
}
};