地上有一个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
方法:dfs
同样是矩阵搜索,可以用深度优先遍历,向一个元素的四个面走,如果符合坐标位数和大于给定的值k,则结果res + 1,走过的元素,声明一个visited(和给定的数组行列数相等)数组做标记,不再走即可。
代码:
class Solution {
boolean[][] visited;
public int movingCount(int m, int n, int k) {
// dfs
this.visited = new boolean[m][n];
return dfs(m, n, 0, 0, k);
}
int dfs(int m, int n, int i, int j, int k) {
if (i < 0 || j < 0 || i >= m || j >= n || cal(i, j) > k || visited[i][j] == true) return 0;
visited[i][j] = true;
// 向四个方向走
// 这里看看了题解,有说明机器人只会向右边和下边走,所以可以不用+向左和上走的dfs可以达到击败100%
return 1 + dfs(m, n, i + 1, j, k) + dfs(m, n, i, j + 1, k) + dfs(m, n, i - 1, j, k) + dfs(m, n, i, j - 1, k);
}
int cal(int i, int j) {
int sum = 0;
while (i != 0) {
sum += (i % 10);
i /= 10;
}
while (j != 0) {
sum += (j % 10);
j /= 10;
}
return sum;
}
}