题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。
例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8> = 19。请问该机器人能够达到多少个格子?
解法一:
很像八皇后问题,通过递归回溯来做,又叫分治法(算法导论P17)
先分析这道题吧。机器人从坐标(0,0)开始移动,移动范围为左,右,上,下。假设机器人从(0,0)向右移动到(0,1),那么它是否还需要再向左移动那?很明显又主动退回去了,而不是通过回溯法回去的。那这样不断的从(0,0)到(0,1),程序会陷入死循环。所以我们可以按从左向右,从上到下的顺序进行递归。
再考虑另一个问题。机器人可以到达的格子。假设有一个1行100列的方格,K为10,当机器人走到(0,29)的时候,还能走到(0,30)吗?虽然,0 + 3 + 0 < 10是成立的,但是它无法到达。所以递归的结束条件除了不能超过方格的边界,还应该再增加一个,当出现一个无法通过的方格时,则向右或向下都是不可达的,此时结束本次递归。
当然,对于其中的每一个格子,走过之后要标记一下,方便最后统计。你也可以选择边走边统计,只要定义一个static变量就好了。
代码如下:
class Solution {
public:
int movingCount(int threshold, int rows, int cols)
{
if (rows <= 0 || cols <= 0)
return 0;
int number = 0;
vector<vector<int>> vec;
for (int i = 0; i < rows; i++)
{
vector<int> v(c