地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:
输入:m = 3, n = 1, k = 0
输出:1
提示:
1 <= n,m <= 100
0 <= k <= 20
分析:
该题目与剑指 Offer 12|| 力扣79. 单词搜索||矩阵中的路径类似,是在此基础上改编的题目,但该题的机器人显然是从[0,0]坐标点出发的,不必再循环找出起始点位,整体的解题的策略依旧是DFS+剪枝
1. 机器人从起始点开始,有两个方向移动,即向下和向右,只需有深度优先遍历列表,[i,j]为当前点位,只需要在保证以下条件下,往一个方向,DFS直至条件不满足
条件:边界不越线 i j在坐标内,点位和不超过K,并且这个点没有记录过
2. 题目有一个细节是相邻的两个格子的坐标和并不一定是连续的
如:10 20 30 ...他们与9 19 29 的坐标和相差8,而其他数值间没有跃迁
i+1 的位数和= (i+1)%10!=0?sum_i+1:sum_i-8
j+1 的位数和= (j+1)%10!=0?sum_j+1:sum_j-8
3. 机器人不能跳跃,只能一个挨着一个走,也就是如果一个格子的左上都为不可达解,这个格子就不能算数
class Solution {
public:
int movingCount(int m, int n, int k) {
//标记计算过的格子
this->m = m;
this->n = n;
vector<vector<bool>> visited(m,vector<bool>(n,false));
return dfs(0,0,k,0,0,visited);
}
private:
int m,n;
int dfs(int i,int j,int k,int sum_i,int sum_j,vector<vector<bool>> &visited){
//判断是否 边框越界 和越界 是否已经计算在内 ,返回无效
if(i>=m||j>=n||(sum_j+sum_i)>k||visited[i][j]==true)return 0;
//当前值有效 标记
visited[i][j] = true;
//当前值有效,加1
int res = 1;
res += dfs(i+1,j,k,(i+1)%10!=0?sum_i+1:sum_i-8,sum_j,visited)
+ dfs(i,j+1,k,sum_i,(j+1)%10!=0?sum_j+1:sum_j-8,visited);
return res;
}
};
当然本题也可以使用广度优先遍历BFS,即不是像DFS一样(一条路按一个方向走到头然后回溯),BFS是一种类似于平推的方法,均匀推进
class Solution {
public:
int movingCount(int m, int n, int k) {
//记录已经走过的点位
vector<vector<bool>> visited(m,vector<bool>(n,false));
int res = 0;
//存放格子的信息
queue<vector<int>> que;
//放入第一个格子的信息,分别是,行,列,行数字和,列数字和
que.push({0,0,0,0});
//迭代
while(!que.empty()){
vector<int> point = que.front();
que.pop();
//提取当前格子的信息
int i = point[0],j =point[1],i_sum = point[2],j_sum = point[3];
//坐标越界,和超过K,已经标记过,直接跳出这次循环
if(i>=m||j>=n||i_sum+j_sum>k||visited[i][j]==true) continue;
//没有越界,当前格可用,标记,res++;
visited[i][j]=true;
++res;
//将当前格的右方和下方,添加到队列中,检测是否满足条件
que.push({i+1,j,(i+1)%10!=0?i_sum+1:i_sum-8,j_sum});
que.push({i,j+1,i_sum,(j+1)%10!=0?j_sum+1:j_sum-8});
}
return res;
}
};