第九章 动态规划part10(代码随想录)

 121. 买卖股票的最佳时机 

1. 确定dp数组(dp table)以及下标的含义

用二维dp数组表示第i天的2种状态

dp[i][0] 表示第i天持有股票所得最多现金,可能i-1天就买股票了

dp[i][1] 表示第i天不持有股票所得最多现金

最后求:dp[len-1][0] dp[len-1][1]

2. 确定递推公式

dp[i][0]由哪些状态推出?

dp[i][0] = dp[i-1][0] 前一天是持有股票最大现金,一直延续持有股票的状态。

-price[i]在第i天买入股票,就变成持有股票的状态。(本题买卖就一次,买肯定是第一次买,直接就是负的)

dp[i][0] = max(dp[i-1][0], -price[i]); 求最大值

--------------------------------------------------------------------------------------------------------------------------------

dp[i][1] = dp[i-1][1] 前一天也可以是不持有股票最大现金,一直延续不持有股票的状态。

dp[i-1][0]+price[i] 前一天持有股票, 在第i天把股票卖了,就变成不持有股票的状态。加上股票价格price[i]。

dp[i][1] = max(dp[i-1][1], dp[i-1][0]+price[i]); 求最大值

3. dp数组如何初始化

递推公式基础:第0天持有 / 不持有股票

dp[0][0] = -prices[0]; // 持有股票的最大现金,买入股票,现金变负数,初始现金为0

dp[0][1] = 0;

4. 确定遍历顺序

从前往后遍历

0的状态已经初始化,i从1开始

for (int i = 1; i < len; i++) {
            dp[i][0] = max(dp[i - 1][0], -prices[i]);
            dp[i][1] = max(dp[i - 1][1], prices[i] + dp[i - 1][0]);
        }

持有股票一定没有不持有股票现金多,输出dp[len-1][i]即可,不用求max(dp[len-1][0], dp[len-1][1])

5. 举例推导dp数组

第九章 动态规划part10(代码随想录)_第1张图片

 

class Solution {
public:
    int maxProfit(vector& prices) {
        int len = prices.size();
        if (len==0) return 0;
        vector> dp(len, vector(2)); //len*2的二维vector
        dp[0][0] -= prices[0];
        dp[0][1] = 0;
        for (int i=1; i

 122.买卖股票的最佳时机II  

股票可以买卖多次。-> 卖出一样,买入不一样了。

持有股票最大现金的状态和上一题不同。

手头现金非0,可能是之前买卖多次获得的利润。

不是0-price[i],是前一天不持有股票的最大现金dp[i-1]减去当天的价格:dp[i - 1][1] - prices[i]

在第i天买入股票,就变成持有股票的状态。

class Solution {
public:
    int maxProfit(vector& prices) {
        int len = prices.size();
        vector> dp(len, vector(2, 0));
        dp[0][0] -= prices[0];
        dp[0][1] = 0;
        for (int i = 1; i < len; i++) {
            dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i]); // 注意这里是和121. 买卖股票的最佳时机唯一不同的地方。
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
        }
        return dp[len - 1][1];
    }
};

你可能感兴趣的:(动态规划,算法,leetcode,数据结构,c++)