剑指 Offer 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)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

终止条件:

  1. 越界
  2. 已经访问过
  3. 超越k值

递推工作:

  1. 将符合条件的i,j添加到visited
  2. 递推顺序下->上->右->左

注意点:

  1. (0,0)特殊值存在,所以返回值最小为1
class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        def getSum(x):
            sum = 0
            while x != 0 :
                sum += x%10
                x = x//10
            return sum
        visited = set()
        res = 0
        def dfs(i, j, k):
            if not 0 <= i < m or not 0 <= j < n or not getSum(i) + getSum(j) <= k or (i, j)  in visited: return False
            else:
                visited.add((i,j))
                return 1 + dfs(i+1, j, k) + dfs(i-1, j, k) + dfs(i, j+1, k) + dfs(i, j-1, k)
        
        for i in range(m):
            for j in range(n):
                return dfs(i, j, k)

除了DFS解法,还有BFS解法,相较于DFS,BFS就是平推,类似于每次三岔路口,有两个选择,两个一起选择,同时走两个方向,在这题表现为,每到一个新的格子,同时选择
递推条件和终止条件和DFS一样。

class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        def getSum(x):
            sum = 0
            while x != 0 :
                sum += x%10
                x = x//10
            return sum

        visited = set()
        from queue import Queue
        q = Queue()
        q.put((0, 0))
        while not q.empty():
            x, y = q.get()
            if (x, y) not in visited and 0 <= x < m and 0 <= y < n and getSum(x) + getSum(y) <= k:
                visited.add((x, y))
                q.put((x+1, y))
                q.put((x, y+1))
        return len(visited)

就是效率比较低……

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值