leetcode.63 - 不同路径II

题目

一个机器人位于一个 *m x n *网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?

网格中的障碍物和空位置分别用 10 来表示。
说明:m 和 *n *的值均不超过 100。
示例 1:

输入: [
  [0,0,0],
  [0,1,0],
  [0,0,0]
]
输出: 2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 `2` 条不同的路径:
1. 向右 -> 向右 -> 向下 -> 向下
2. 向下 -> 向下 -> 向右 -> 向右

思路

个人思路就是在62题动态规划方法上的改进
完成后也参考了一些其他同学的答案,貌似大都如此

代码

class Solution:
    def uniquePathsWithObstacles(self, obstacleGrid: List[List[int]]) -> int:
        
        n = len(obstacleGrid)
        m = len(obstacleGrid[0])
        
        if((obstacleGrid[0][0] == 1) or (obstacleGrid[n-1][m-1] == 1)):
            return 0
        
        row1 = [1]
        for i in range(1,m):
            if(obstacleGrid[0][i] == 1):
                row1.append(0)
            else:
                row1.append(row1[i-1])
        
        row_num=1
        while(row_num < n):
            if(obstacleGrid[row_num][0] == 1):
                row2 = [0]
            else:
                row2 = [row1[0]]
            
            for i in range(1,m):
                if(obstacleGrid[row_num][i] == 1):
                    row2.append(0)
                else:
                    row2.append(row2[i-1]+ row1[i])
                
            row1 = row2
            row_num+=1
        return row1[-1]

欢迎大家关注我的公众号


半亩房顶

你可能感兴趣的:(leetcode.63 - 不同路径II)