代码随想录算法训练营第四十九天| 121.买卖股票的最佳时机、122.买卖股票的最佳时机 II

121 买卖股票的最佳时机

题目链接：121
思路：定义两个状态：

• 0：不持有股票。
• 1：持有股票。

则dp[i][j]代表第i天时状态为j时，手上才持有的最大现金。持有股票时，现金可以为负。对两个状态进行分析：

• 不持有时，有两种情况：昨天也不持有，则今天继承昨天的dp[i-1][0]；昨天持有，今天不持有，即今天卖出，则现金为dp[i-1][1]+prices[i]。因此，dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i])。
• 持有时，有两种情况：昨天也持有，今天继承昨天dp[i-1][1]；昨天不持有，今天持有，即今天买入，由于只能买一次，此前现金一定为0，则今天的现金为-prices[i]。因此，dp[i][1]=max(dp[i-1][1],-prices[i])。

代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        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][0], dp[i-1][1]+prices[i]);
            dp[i][1] = max(dp[i-1][1], -prices[i]); 
        }
        return dp.back()[0];
    }
};

122 买卖股票的最佳时机 II

题目链接：122
思路：与121 买卖股票的最佳时机类似，但由于可以无限次买入买出，因此当昨天不持有今天持有时，即今天买入时，可以继承昨天剩下的现金，dp[i][1]=dp[i-1][0]-prices[i]。
代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
       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][0], dp[i-1][1]+prices[i]);
           dp[i][1] = max(dp[i-1][1], dp[i-1][0]-prices[i]);
       } 
       return dp.back()[0];
    }
};