这道题目,代码随想录上已经给了非常全面的解答,
存在的问题是,初始化数组的时候也要用函数传递,比如第一行中间有一个障碍,那么后面的元素就传递不过去了。
错误初始化代码:
for i in range(col):
if obstacleGrid[0][i]==0:
dp[0][i]=1
else:
dp[0][i]=0
for j in range(row):
if obstacleGrid[j][0]==0:
dp[j][0]=1
print(j)
else:
dp[j][0]=0
正确初始化代码:
dp[0][0] = 1 if obstacleGrid[0][0] != 1 else 0
if dp[0][0] == 0: return 0 # 如果第一个格子就是障碍,return 0
# 第一行
for i in range(1, col):
if obstacleGrid[0][i] != 1:
dp[0][i] = dp[0][i-1]
# 第一列
for i in range(1, row):
if obstacleGrid[i][0] != 1:
dp[i][0] = dp[i-1][0]
还有一个问题初始化二维数组,千万不要[[0]*n]*m,改动一个元素会改变一整列元素,应该用dp=[[0 for _ in range(col)] for p in range(row)]方式初始化数组
整体代码
class Solution:
def uniquePathsWithObstacles(self, obstacleGrid: List[List[int]]) -> int:
if obstacleGrid[0][0]==1:
return 0
row=len(obstacleGrid)
col=len(obstacleGrid[0])
dp=[[0 for _ in range(col)] for p in range(row)]
dp[0][0]=1
for i in range(1,col):
if obstacleGrid[0][i]==0:
dp[0][i]=dp[0][i-1]
for j in range(1,row):
if obstacleGrid[j][0]==0:
dp[j][0]=dp[j-1][0]
print(dp)
if row<=1 or col<=1:
return dp[row-1][col-1]
for i in range(1,col):
for j in range(1,row):
if obstacleGrid[j][i]==1:
dp[j][i]=0
else:
dp[j][i]=dp[j-1][i]+dp[j][i-1]
return dp[row-1][col-1]