第九章 动态规划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数组

 

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        if (len==0) return 0;
        vector<vector<int>> dp(len, vector<int>(2)); //len*2的二维vector
        dp[0][0] -= prices[0];
        dp[0][1] = 0;
        for (int i=1; i<len; i++){
            // 前一天是持有股票最大现金，一直延续持有股票的状态
            // 在第i天买入股票，变成持有股票的状态
            dp[i][0]=max(dp[i-1][0], -prices[i]); 
            // 前一天可以是不持有股票最大现金，一直延续不持有股票的状态
            // 前一天持有股票, 在第i天把股票卖了，就变成不持有股票的状态。
            dp[i][1]=max(dp[i-1][1], dp[i-1][0]+prices[i]);
        }
        return dp[len-1][1];
    }
};

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

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

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

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

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

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

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        vector<vector<int>> dp(len, vector<int>(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];
    }
};