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)