题目[题目及部分解法来源于LeeCode]
地上有一个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。请问该机器人能够到达多少个格子?
solution one:
呐呐呐,有木有感觉到似曾相识???[剑指offer刷题12] - 矩阵中的路径,其实和本题差不多。因为基本上都是在二维数组中进行搜索。那有木有想到深度优先搜索嘞?
这一题相当于我们在一个矩阵中找到合适的点,求得总共有多少点即可。看思路:
- 第一点:起点的选择
- 在矩阵中的路径题目中,需要的是搜索起点,所以在代码一开始双重for循环来调用dfs,当不是起点值的时候返回二重for循环,再次调用dfs以查找匹配起点。
- 本题中,由于给定从坐标[0,0]开始探测移动范围,故不需要进行对起点的搜索。直接调用dfs。
- 第二点:本题递归返回的条件
- 需要满足机器人走的每一步都要在矩阵中
- 行列坐标各数位之和必须小于等于K
- 如果满足上述两条,则机器人可走的格子加一,即res++;
- 满足上述2点,我们进行下一步查找。即可以从上、下、左、右进行查找,并通过标记来进行剪枝操作。
class Solution {
//计格器
public int res = 0;
public int movingCount(int m, int n, int k) {
// boolean数组用于剪枝操作
boolean[][] arr = new boolean[m][n];
//不需要查找起点,直接调用
dfs(0, 0, m, n, k, arr);
return res;
}
public void dfs(int i, int j, int m, int n, int k, boolean[][] arr){
// 判断是否满足递归返回条件及是否走过此点
if( i < 0 || i >= m || j < 0 || j >= n || arr[i][j] ) return;
//没有走过,将其标记
arr[i][j] = true;
//计算sum,判断是否小于等于K
int sum = i % 10 + j % 10 + i / 10 + j / 10;
if(sum > k) return;
//符合题意,机器人可走格数量加一
++res;
//上下左右四个方向搜索可走点
dfs(i + 1, j, m, n, k, arr);
dfs(i - 1, j, m, n, k, arr);
dfs(i, j + 1, m, n, k, arr);
dfs(i, j - 1, m, n, k, arr);
}
}