有意思的算法题小记。
1. 题目
@地上有一个m行和n列的方格。一个机器人从坐标x,y的格子开始移动,
@ 每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。
@例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),
@因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
2. 思路
这道题目可以使用动态规划来解答。动态规划问题的思考方式是,当前状态和前一个状态,后一个状态的关联有什么递进关系。下面提供一下我的思考方式。
目的是从一点开始,一直探索到最大范围。设定中间某一时刻为当前状态,他和前一个状态的差别是最少多了一个有效的格子,和后一个状态也类似。归纳出来就是已有很多格子,还要从当前的格子身上去探索更多的格子。由题目可知,每个格子只能向四个方向探索一格,所以手中的格子可以分为两种,一种是确定不能向外探索的,一种是未知。 由此可以推导出,从当前状态出发,只要遍历到未知的数量为0,即不能再向外探索。当然,再遍历过程中,未知的数量是可以增加的。
3. 代码实现参考
#include <iostream>
#include <queue>
#include <vector>
static bool isValid(int x, int y, int k) {
int sum = 0;
while (x>0) {
sum += x%10;
x/=10;
}
while (y>0) {
sum += y%10;
y/=10;
}
if (sum > k) return false;
return true;
}
// 输入 m行n列的方格,提供初始位置(x,y), 以及限制量k
// 返回 可以探索到格子的数量
int func(int m, int n, int x, int y, int k) {
// 坐标有效判断,即默认边界判断,非法返回true
auto isIllegalCoord = [&](int _x, int _y) -> bool { return !(_x >=0 && _y >= 0 && _x < n && _y < m);};
// visited 区分是否是遍历过的格子
std::vector<std::vector<bool>> visited(m, std::vector<bool>(n, false));
std::queue<std::pair<int,int> > coords; // 保存还没有遍历过的格子
int step[4][2] = {{0,-1}, {0,1}, {-1, 0}, {1, 0}};
// 设置初始值
int ans = 0;
if (isValid(x, y, k) == true) {
coords.push(std::make_pair(x, y));
visited[x][y] = true;
ans++;
}
// 直到遍历到未知的格子为空
while (!coords.empty()) {
// 取出队列最前端的格子
std::pair<int, int> tmp = coords.front();
coords.pop();
for (int i = 0; i < 4; i++) {
std::pair<int, int> cur = {tmp.first + step[i][0], tmp.second + step[i][1]};
// 如果坐标非法或已经遍历过,则跳过
if (isIllegalCoord(cur.first,cur.second) || visited[cur.first][cur.second] == true)
continue;
// 满足条件,则放入coords中,等待再从此点上继续扩展
if (true == isValid(cur.first, cur.second, k)) {
coords.push(cur);
visited[cur.first][cur.second] = true;
ans++;
}
}
}
std::cout << "ans: " << ans << std::endl;
return ans;
}