代码随想录算法训练营第五十二天 | LeetCode 121、122

---

前言

LeetCode题目：LeetCode 121、122
Takeaway：银行买卖股票问题，主要难度在定义DP数组上，理解 0 和 1的区别，持有并不是仅仅指刚买股票 i ，如果是之前就买过了，现在还持有，也算是持有；

---

一、121

持有（0）和不持有（1）股票分别最多有多少钱。持有并不是仅仅指刚买股票 i ，如果是之前就买过了，现在还持有，也算是持有；不持有也不是仅仅指刚买，也可以是以前就卖了。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        // 定义DP数组，DP含义是第 i 天，持有（0）和不持有（1）股票分别最多有多少钱
        vector<vector<int>> dp(prices.size(), vector<int>(2));
        
        dp[0][0] = -prices[0];
        dp[0][1] = 0;

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

        return max(dp[prices.size()-1][0], dp[prices.size()-1][1]);
    }
};

---

二、122

上一道题的变体类型题，只有dp公式发生了一些小改动，整体区别不大。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        // DP数组含义：0不持有股票，1持有股票
        vector<vector<int>> dp(prices.size(), vector<int>(2));
        dp[0][0] = 0;
        dp[0][1] = -prices[0];

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

        return max(dp[prices.size()-1][0], dp[prices.size()-1][1]); 
    }
};

---

总结

银行买卖股票问题，主要难度在定义DP数组上，理解 0 和 1的区别，持有并不是仅仅指刚买股票 i ，如果是之前就买过了，现在还持有，也算是持有。