剑指 Offer 13. 机器人的运动范围(分治,python注意二维数组定义容易踩浅拷贝的坑)

https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof/
题目:地上有一个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。请问该机器人能够到达多少个格子?

思路:
分制思想:当前能遍历到节点数等于当前节点相邻所有节点能遍历到的节点数总和,注意重复遍历节点只记录一次,因此需要用另一个相同维度矩阵记录该节点是否已被计数。因为是从0,0左上角开始,只需要往row+1和col+1两个方向遍历即可。
算法时间复杂度和空间复杂度应该都是O(mn)

第一版代码,提交错误,计算出结果总是比总数少。原因,使用[[False]*n]*m定义了flag矩阵,这个定义的含义其实是将[False]*n]这个list浅拷贝m次,当更改flag[i][j]元素时,所有行的第i个元素都会被更改。

class Solution(object):
    def dfs(self, row, col, m, n, flag, k):
        if row < 0 or row >= m or col < 0 or col >= n or (row / 10 + row % 10 + col / 10 + col % 10) > k or flag[row][col]:
            return 0
        print "flags1: " + str(flag)   #   flags1: [[False, False], [False, False], [False, False]]
        print str(row) + "\t" + str(col)   # 0	0
        flag[row][col] = True  
        print "flags2: " + str(flag)    # flags2: [[True, False], [True, False], [True, False]]
        return self.dfs(row + 1, col, m, n, flag, k) + self.dfs(row, col + 1, m, n, flag, k) + 1
            
    def movingCount(self, m, n, k):
        """
        :type m: int
        :type n: int
        :type k: int
        :rtype: int
        """
        flag = [[False]*n]*m
        return self.dfs(0, 0, m, n, flag, k) 

将flag定义更改为flag = [[False] * n for i in range(m)]

class Solution(object):
    def dfs(self, row, col, m, n, flag, k):
        if row < 0 or row >= m or col < 0 or col >= n or (row / 10 + row % 10 + col / 10 + col % 10) > k or flag[row][col]:
            return 0
        flag[row][col] = True
        return self.dfs(row + 1, col, m, n, flag, k) + self.dfs(row, col + 1, m, n, flag, k) + 1
            
    def movingCount(self, m, n, k):
        """
        :type m: int
        :type n: int
        :type k: int
        :rtype: int
        """
        # flag = [[False]*n]*m
        flag = [[False] * n for i in range(m)]
        return self.dfs(0, 0, m, n, flag, k) 

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值