代码随想录算法训练营第三十九天| 动态规划

Posted by Zhuoqun on February 21, 2023

62.不同路径

这道题之前做过,有印象,所以自己能写出来。但是在2d vector初始化的时候写的格式不对。其实可以直接用2d int来做的,不一定非要用vector。思维僵化了。

class Solution {
public:
    int uniquePaths(int m, int n) {
        vector<vector<int>> dp(m,vector<int> (n));
        dp[0][0] = 1;
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if(i==0 || j ==0){
                    dp[i][j] = 1;
                }
            }
        }
        for(int i=1; i<m; i++){
            for(int j=1; j<n; j++){
                dp[i][j] = dp[i-1][j] + dp[i][j-1];
            }
        }
        return dp[m-1][n-1];

    }
};

63. 不同路径 II

和上一题是一个思路的,我写的有点冗余。(我发现写的和之前几乎是一模一样)

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        int m = obstacleGrid.size();
        int n = obstacleGrid[0].size();
        vector<vector<int>>dp(m, vector<int>(n));
        bool findObstacle = false;
        for(int i=0; i<m; i++){
            if(findObstacle){
                dp[i][0] = 0;
            }
            else{
                if(obstacleGrid[i][0] == 1){
                    dp[i][0] = 0;
                    findObstacle = true;
                }
                else{
                    dp[i][0] = 1;
                }
            }
        }
        findObstacle = false;
        for(int j=0; j<n; j++){
            if(findObstacle){
                dp[0][j] = 0;
            }
            else{
                if(obstacleGrid[0][j] == 1){
                    dp[0][j] = 0;
                    findObstacle = true;
                }
                else{
                    dp[0][j] = 1;
                }
            }
        }
        for(int i=1; i<m; i++){
            for(int j=1; j<n; j++){
                if(obstacleGrid[i][j] == 1){
                    dp[i][j] = 0;
                }
                else{
                    dp[i][j] = dp[i-1][j] + dp[i][j-1];
                }
            }
        }
        return dp[m-1][n-1];
    }
};

根据Carl的题解,我精简了一下我的代码,这样看起来就整洁多了。

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        int m = obstacleGrid.size();
        int n = obstacleGrid[0].size();
        if (obstacleGrid[m - 1][n - 1] == 1 || obstacleGrid[0][0] == 1) 
            return 0;
        vector<vector<int>>dp(m, vector<int>(n,0));
        for(int i=0; i<m && obstacleGrid[i][0]==0; i++){
            dp[i][0] = 1;
        }
        for(int j=0; j<n && obstacleGrid[0][j]==0; j++){
            dp[0][j] = 1;
        }
        for(int i=1; i<m; i++){
            for(int j=1; j<n; j++){
                if(obstacleGrid[i][j] == 1){
                    dp[i][j] = 0;
                }
                else{
                    dp[i][j] = dp[i-1][j] + dp[i][j-1];
                }
            }
        }
        return dp[m-1][n-1];
    }
};