[LeetCode] 动态规划入门题目

最近接触了动态规划这个厉害的方法,还在慢慢地试着去了解这种思想,因此就在LeetCode上面找了几道比较简单的题目练了练手。

首先,动态规划是什么呢?很多人认为把它称作一种“算法”,其实我认为把它称作一种“思想”更为合适;利用动态规划去解决问题,其实就是逐步递推的过程,与贪心算法不同,动态规划递推的每一步都要求是当前的最优解(这是很重要的,递推的正确性依赖的就是这一点);利用动态规划解题时,必须自己定义出来状态和状态转移方程。然而,看上去简单,做起来却非常困难,因为解题时的具体形式千差万别,找出问题的子结构以及通过子结构重新构造最优解的过程很难统一。

经典的动态规划题目有背包问题、硬币问题等等,可以通过这些题目去理解一下这个东西。

我认为,动态规划最难的就是找出状态方程。同时,个人认为比较难理解的一点是,懂得“前面每一步都是最优解”这个前提。

废话不多说,直接看看LeetCode上简单的动态规划题目。

要注意的是,下面的三题都用到了局部最优和全局最优解法:

1.Jump Game

原题地址:https://leetcode.com/problems/jump-game/description/

解法:

用一个global变量保存到目前为止能跳的最远距离,用一个local变量保存当前一步出发能跳的最远距离,这题里面的状态就是走到每一步时的global[i]值,状态转移方程就是global[i] =max{nums[i] + i, global[i-1]}。当然,写代码的时候用变量代替数组即可。

class Solution {
public:
    bool canJump(vector<int>& nums) {
      int reach = 0;
      for (int i = 0; i < nums.size() - 1 && reach >= i; i++) {
          reach = nums[i] + i > reach ? nums[i] + i : reach;
      }
      return reach >= nums.size() - 1;
  }
};

 

2.Maximum Subarray

原题地址:https://leetcode.com/problems/maximum-subarray/description/

解法:

这一题要维护两个变量:global和local,与上面一题一样,local保存包含当前元素的最大值(局部最优),global保存的是所有情况里面的最大值(全局最优)。假设第i步的local[i]和global[i]已知,那么第i+1步的local[i + 1] = max{ nums[i] + local[i], nums[i + 1] },global[i + 1]  = max{global[i], local[i + 1]}。代码如下:

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
      int global = nums[0], local = nums[0];
      for (int i = 1; i < nums.size(); i++) {
          local = nums[i] > nums[i] + local ? nums[i] : nums[i] + local;
          global = local > global ? local : global;
      }       
      return global;
  }
};

 

3.Best Time to Buy and Sell Stock

原题地址:https://leetcode.com/problems/best-time-to-buy-and-sell-stock/description/

这道题目有两种方法,其实都是动态规划:

(1)

class Solution {
public:
   int maxProfit(vector<int>& prices) {
       if (prices.size() == 0) return 0; 
      int maxPrice = prices[prices.size() - 1];
      int res = 0;
      for (int i = prices.size() - 1; i >= 0; i--) {
          maxPrice = max(maxPrice, prices[i]);
          res = max(res, maxPrice - prices[i]);
      }
      return res;     
  }
};

这种解法在这个博客里面讲得很详细:http://www.cnblogs.com/remlostime/archive/2012/11/06/2757434.html

(2)局部最优和全局最优解法:

class Solution {
public:
   int maxProfit(vector<int>& prices) {
      if (prices.size() == 0) return 0;
      int local = 0, global = 0;
      for (int i = 1; i < prices.size(); i++) {
          local = max(0, local + prices[i] - prices[i - 1]);
          global = max(local, global);
      }
      return global;   
  }
};
local = max(0, local + prices[i] - prices[i - 1])这一句,我一开始在考虑:为什么不写成local = max(local, local + prices[i] - prices[i - 1])呢?
后来想了一下,因为假如这样写,有可能得到的就不是包含当前元素的局部最优解了。所以,在“局部最优和全局最优解法”里面,永远不会出现local=local的情况。


4.Minimum Path Sum
原题地址:https://leetcode.com/problems/minimum-path-sum/description/
这道题目不需用到上面的“局部最优和全局最优”解法,只需要每次选出最优的即可。除了边界的元素,其他元素的最优都是 min{min[i - 1][j] + grid[i][j],min[i][j - 1] + grid[i][j]
}。
class Solution {
public:
    int minPathSum(vector<vector<int>>& grid) {
        int ** min  = new int*[grid.size()];
        for (int i = 0; i < grid.size(); i++) {
            min[i] = new int[grid[i].size()];
        }
        min[0][0] = grid[0][0];
        for (int i = 0; i < grid.size(); i++) {
            for (int j = 0; j < grid[i].size(); j++) {
                if (i == 0 && j == 0) continue;
                else if (i == 0) min[i][j] = min[i][j - 1] + grid[i][j];
                else if (j == 0) min[i][j] = min[i - 1][j] + grid[i][j];
                else min[i][j] = min[i - 1][j] + grid[i][j] < min[i][j - 1] + grid[i][j] ? min[i - 1][j] + grid[i][j] : min[i][j - 1] + grid[i][j] ;
            }
        }
        return min[grid.size() - 1][grid[grid.size() - 1].size() - 1];
        return 0;
    }
};


5.Triangle
地址:https://leetcode.com/problems/triangle/description/

也是一道典型的dp题目,思想跟上面一题差不多:
class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        if (triangle.size() == 1) return triangle[0][0];
        int ** min = new int *[triangle.size()];
        for (int i = 0; i < triangle.size(); i++) {
            min[i] = new int[triangle[i].size()];
        }
        min[0][0] = triangle[0][0];
        int res = INT_MAX;
        for (int i = 0; i < triangle.size(); i++) {
            for (int j = 0; j < triangle[i].size(); j++) {
                if (i == 0 && j == 0) continue;
                else if (j == 0) min[i][j] = min[i - 1][j] + triangle[i][j];
                else if (j == triangle[i].size() - 1) min[i][j] = min[i - 1][j - 1] + triangle[i][j];
                else min[i][j] = min[i - 1][j - 1] + triangle[i][j] < min[i - 1][j] + triangle[i][j] ? min[i - 1][j - 1] + triangle[i][j] : min[i - 1][j] + triangle[i][j];
                if (min[i][j] < res && i == triangle.size() - 1) {
                    res = min[i][j];
                }
            }
        }
        return res;
    }
};
但这道题有趣的地方在于,空间复杂度可以缩小到O(n):我们把这个三角形倒过来看,便能发现可以通过复用一个一维数组来储存最小值:
class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        vector<int> min = triangle[triangle.size() - 1];
        for (int i = triangle.size() - 2; i >= 0; i--) {
            for (int j = 0; j <= i; j++) {
                min[j] = min[j] < min[j + 1] ? min[j] + triangle[i][j] : min[j + 1] + triangle[i][j];
            }    
        }
        return min[0];
    }
};

 








posted @ 2017-09-29 21:40  fengzw  阅读(4272)  评论(0编辑  收藏  举报