题目

一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为 “Start” )。

机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为 “Finish”)。

现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?

网格中的障碍物和空位置分别用 1 和 0 来表示。

示例

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

  1. 向右 -> 向右 -> 向下 -> 向下
  2. 向下 -> 向下 -> 向右 -> 向右
解析

这道题和上一道相比,就是多了一个障碍物;
首先五部曲的思路是不能变的,但初始化条件有所更改,比如在第一行上,若有障碍物,则从该位置开始,右边的都走不到了,也是是默认的初始化为0;

func uniquePathsWithObstacles(obstacleGrid [][]int) int {
    m, n := len(obstacleGrid), len(obstacleGrid[0])
    dp := make([][]int, m)
    //里面的一维数组初始化得提前到这了
    for i, _ := range dp {
        dp[i] = make([]int, n)
    }
    //初始化, 如果是障碍物, 后面的就都是0, 不用循环了
    for i := 0; i < m && obstacleGrid[i][0] == 0; i++ {
        dp[i][0] = 1
    }
    for i := 0; i < n && obstacleGrid[0][i] == 0; i++ {
        dp[0][i] = 1
    }

    for i := 1; i < m; i++ {
        for j := 1; j < n; j++ {
            if obstacleGrid[i][j] != 1 { //如果obstacleGrid[i][j]这个点是障碍物, 那么dp[i][j]保持为0
                dp[i][j] = dp[i-1][j] + dp[i][j-1]
            }
        }
    }
    return dp[m-1][n-1]
}

再次注意下,dp[i][j]是表示从(0 ,0)出发,到(i, j) 有dp[i][j]条不同的路径

12-06 05:19