码迷,mamicode.com
首页 > 其他好文 > 详细

动态规划之 minimum path sum

时间:2016-02-20 01:58:48      阅读:221      评论:0      收藏:0      [点我收藏+]

标签:动态规划

Given a m x n grid filled with non-negative numbers, find a path from top left to bottom right which minimizes the sum of all numbers along its path.

Note: You can only move either down or right at any point in time.

假定gird用于保存m*n数字,

dp[i][j]表示到第i行第j列的最小路径和,则每次往右或者往下走,则得到递推公式

dp[i][j]=min(dp[i-1][j],dp[i][j-1])+grid[i][j]


初始值:

i=0,j=0  dp[i][j]=grid[i][j];

i=0,j>0  此时只能从左往右走,则dp[i][j]=dp[i][j-1]+grid[i][j];

i>0,j=0  此时只能从上往下走,则dp[i][j]=dp[i-1][j]+grid[i][j];


int minPathSum(vector<vector<int>>& grid) {

        int m=grid.size();

        if(m==0)

            return -1;

        int n=grid[0].size();

        

        vector<vector<int> > dp(m,vector<int>(n));

        

        for(int i=0;i<m;i++){

            for(int j=0;j<n;j++){

                if(i==0){

                    if(j==0){

                        dp[i][j]=grid[i][j];

                    }

                    else{

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

                    }

                }

                else{

                    if(j==0){

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

                    }

                    else{

                        dp[i][j]=min(dp[i-1][j],dp[i][j-1])+grid[i][j];

                    }

                }

            }

        }

        return dp[m-1][n-1];

    }


优化空间

用一维数组代替二维数组,

由dp[i][j]=min(dp[i-1][j],dp[i][j-1])+grid[i][j] 知

dp[i][j]和dp[i-1][j],dp[i][j-1],grid[i][j]有关,这时如果计算顺序是正向,下标从小开始,

则在第i次循环,计算dp[j] 时,dp[i-1][j]可以看作是dp[j]的旧值,即第i-1次循环时的值,此时已经计算完成,dp[i][j-1]可以看作dp[j-1]

则可以更新为

dp[j]=min(dp[j-1],dp[j])+grid[i][j]


在代码中直接把dp[i][j] ==>dp[j]  dp[i-1][j]==>dp[j]

  vector<int> dp(n);

        

        for(int i=0;i<m;i++){

            for(int j=0;j<n;j++){

                if(i==0){

                    if(j==0){

                        dp[j]=grid[i][j];

                    }

                    else{

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

                    }

                }

                else{

                    if(j==0){

                        dp[j]=dp[j]+grid[i][j];

                    }

                    else{

                        dp[j]=min(dp[j],dp[j-1])+grid[i][j];

                    }

                }

            }

        }

        

        return dp[n-1];

动态规划之 minimum path sum

标签:动态规划

原文地址:http://searchcoding.blog.51cto.com/1335412/1743488

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!