【leetcode】63 不同路径II（动态规划）

题目链接：https://leetcode-cn.com/problems/unique-paths-ii/

题目描述

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

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

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

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

说明：m 和 n 的值均不超过 100。

示例 1:

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

思路

典型的动态规划题目，同62题解法；我们用一个二维数组dp来存储当前位置的不同路径数。

动态规划方程为：

dp[i][j] = dp[i-1][j] + dp[i][j-1]

区别在于如果obstacleGrid[i][j] == 1，则直接令dp[i][j]=0

注意：重点在于C++用int必然会越界。怎么改都越界，要强转成（long )

复杂度分析
时间复杂度：O（mn）
空间复杂度：O（mn）
m，n分别为矩阵行列数

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        if(obstacleGrid.empty() || obstacleGrid[0].empty()) return 0;
        int rows = obstacleGrid.size(), cols = obstacleGrid[0].size();
        if (obstacleGrid.back().back() == 1 ||obstacleGrid[0][0] == 1)
            return 0;

        vector<vector<long >> dp(rows, vector<long >(cols, 0));
        dp[0][0] = 1;
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if(obstacleGrid[i][j] == 1){
                    dp[i][j] = 0;
                    continue;
                }
                if(i==0 && j==0) continue;
                long up = 0, left = 0;
                if(i-1>=0) up = dp[i-1][j];
                if(j-1>=0) left = dp[i][j-1];
                dp[i][j] = up+left;
            }
        }
        return dp.back().back();
    }
};

进一步优化，我们可以使用obstacleGrid 作为dp数组，将空间复杂度降为O（1）

/*
 * 动态规划，使用obstacleGrid数组作为dp数组
 * 时间复杂度O(N^2) 空间复杂度O(1)
 */
class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        if(obstacleGrid.empty() || obstacleGrid[0].empty()) return 0;
        int rows = obstacleGrid.size(), cols = obstacleGrid[0].size();
        if (obstacleGrid.back().back() == 1 ||obstacleGrid[0][0] == 1)
            return 0;

        obstacleGrid[0][0] = 1;
        // 第一列元素
        for (int i = 1; i < rows; ++i)
            obstacleGrid[i][0] = (obstacleGrid[i][0] == 0 && obstacleGrid[i-1][0] == 1 ) ? 1:0; // 设置为1表示有1条路径
        // 第一行元素
        for (int j = 1; j < cols; ++j)
            obstacleGrid[0][j] = (obstacleGrid[0][j] == 0 && obstacleGrid[0][j-1] == 1)? 1:0;   // 设置为1表示有1条路径


        for (int i = 1; i < rows; ++i) {
            for (int j = 1; j < cols; ++j) {
                if(obstacleGrid[i][j] == 0)
                    obstacleGrid[i][j] = obstacleGrid[i-1][j] + obstacleGrid[i][j-1];
                else
                    obstacleGrid[i][j] = 0;
            }
        }
        return obstacleGrid.back().back();
    }
};