题目描述:
地上有一个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。请问该机器人能够到达多少个格子? 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例1:
输入:m = 2, n = 3, k = 1 输出:3
示例2:
输入:m = 3, n = 1, k = 0 输出:1
提示:
1 <= n,m <= 100 0 <= k <= 20
题目难度:中等
分析:
起初没理解题意,我以为只是遍历二维数组,然后分别判断方格中的数是否大于k值而已,后来提交失败,才明白因为会出现很多不可达的情况然后就会阻断机器人的前进路线,形成障碍物的效果。所以此题应该用DFS(深度优先遍历)来解。代码如下:
java:
class Solution { // 模拟所有的方格 boolean[][] visited; public int movingCount(int m, int n, int k) { // m * n的方格 visited = new boolean[m][n]; // 最开始在[0, 0]的位置上 return dfs(0, 0, m, n, k); } private int dfs(int i, int j, int m, int n, int k) { // 定义边界(剪枝) if (i >= m || j >= n || visited[i][j] || (i / 10 + i % 10 + j / 10 + j % 10) > k) { return 0; } // 走过的方格设置为true,过滤重复 visited[i][j] = true; // 1:代表此方格可以计数;i + 1,j + 1:代表机器人可以向下或向右行走 return 1 + dfs(i + 1, j, m, n, k) + dfs(i, j + 1, m, n, k); } }
python:
class Solution: def movingCount(self, m: int, n: int, k: int) -> int: visited = [[False for _ in range(n)] for _ in range(m)] return self.dfs(0, 0, m, n, k, visited) def dfs(self, i: int, j: int, m: int, n: int, k: int, visited: List[List[bool]]) -> int: if i >= m or j >= n or visited[i][j] or (i // 10 + i % 10 + j // 10 + j % 10) > k: return 0 visited[i][j] = True return 1 + self.dfs(i + 1, j, m, n, k, visited) + self.dfs(i, j + 1, m, n, k, visited)
另外python也可以这么写,但是我个人喜欢上面的那种,因为看着比较清晰。
class Solution: def movingCount(self, m: int, n: int, k: int) -> int: visited = [[False for _ in range(n)] for _ in range(m)] def dfs(i: int, j: int) -> int: if i >= m or j >= n or visited[i][j] or (i // 10 + i % 10 + j // 10 + j % 10) > k: return 0 visited[i][j] = True return 1 + dfs(i + 1, j) + dfs(i, j + 1) return dfs(0, 0)
总结:
时间复杂度为O ( m n ) ,DFS:深度优先遍历。很有代表性的题目,一定要掌握。