面试题13. 机器人的运动范围
地上有一个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
(来源:力扣(LeetCode))
思路:
矩阵搜索问题,使用深度优先搜索算法解决。
题目要求的是遍历行坐标和列坐标的数位之和小于k的格子:
1、将x % 10,得到x的个位数;
2、将x / 10,令 x 的十进制数向右移动一位,即删除个位数字。
遍历矩阵时,因为机器人从(0,0)开始移动,每次只能移动一格,所以只能通过向右和向下移动,访问所有可达解。
DFS 通过递归,先朝一个方向搜到底,再回溯至上个节点,沿另一个方向搜索,以此类推。
递归终止的条件:
(1) 行列索引越界 ;
(2) 数位和超出目标值 k ;
(3) 当前元素已访问过 ,返回 0 ,代表不计入可达解。
我们设置一个boolean数组visited,表示元素已经被访问过了。接着 计算当前元素的 下、右 两个方向元素的数位和,并开启下层递归。
返回值: 返回 1 + 下方搜索的可达解总数 + 右方搜索的可达解总数,代表从本单元格递归搜索的可达解总数。
例如示例1:
输入:m = 2, n = 3, k = 1
输出:3
(0,0) | (0,1) | (0,2) |
---|---|---|
(1,0) | (1,1) | (1,2) |
机器人从(0,0)开始搜索。
(0,1) | (0,2) | |
---|---|---|
(1,0) | (1,1) | (1,2) |
dfs的顺序是下、右,所以先向下移动。
(0,1) | (0,2) | |
---|---|---|
(1,1) | (1,2) |
对于单元格(1,0)来说:
向下搜索,则索引越界,返回0;
向右搜索,则位数和大于k,返回0。
回溯至(0,0),向右搜索,(0,1)为可达解。
(0,2) | ||
---|---|---|
(1,1) | (1,2) |
对于单元格(0,1):
向下搜索,则数位和大于k,返回0;
向右搜索,则数位和大于k,返回0。
返回数值为1 + 1 + 0 + 0 + 1 + 0 + 0 = 3
class Solution {
boolean[][] visited ;
public int movingCount(int m, int n, int k) {
visited = new boolean[m][n];
return dfs(0, 0, m, n, k);
}
public int dfs(int x, int y, int m, int n, int k) {
if ( x < 0 || x >=m || y < 0 ||y >= n
|| (x % 10 + x / 10 + y % 10 + y /10) > k || visited[x][y]) {
return 0;
}
visited[x][y] = true;
return 1 + dfs(x + 1, y, m, n, k) + dfs(x, y + 1, m, n, k);
}
}