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

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

确定dp数组：

dp[i][j] : 第i天所剩余的金钱为dp[i][j]

j : 2种状态：

j = 0 : 买入
j = 1 : 卖出

确定dp转移方程：

可以达到dp[i][0]的操作：
		 前一天是买入 : dp[i][0] = dp[i-1][0]
		 前一天是卖出 : dp[i][0] = dp[i-1][1] - prices[i]
		 
可以达到dp[i][1]的操作：
		 前一天是买入 : dp[i][1] = dp[i-1] + prices[i]
		 前一天是卖出 : dp[i][1] = dp[i-1][1]

确定dp初值：

dp[0][0] = -prices[0]
dp[0][1] = 0

完整代码：

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

LeetCode 123. 买卖股票的最佳时机 III

j的状态：

j = 0 : 无操作    // 是必要的
j = 1 : 第一次买入
j = 2 : 第一次卖出
j = 3 : 第二次买入
j = 4 : 第二次卖出

dp状态转移方程：

可以达到 j = 0 的操作：
		前一天是无操作

可以达到 j = 1 的操作：
		前一天就是第一次买入
		前一天是无操作
		
可以达到 j = 2 的操作：
		前一天是第一次买入
		前一天是第一次卖出

可以达到 j = 3 的操作：
		前一天是第一次卖出
		前一天就是第二次买入
		
可以达到j = 4 的操作：
		前一天是第二次买入
		前一天就是第二次卖出

确定dp初值

dp[0][0] = 0
dp[0][1] = -prices[0]
dp[0][2] = 0
dp[0][3] = -prices[0]  //!!!
dp[0][4] = 0

	0	1	2	3	4
7	0	-7	0	-7	0
6	0	0
4	0
3	0
1	0

完整代码;

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(),vector<int>(5,0));
        dp[0][1] = -prices[0];
        dp[0][3] = -prices[0];
        for(int i=1;i<prices.size();i++)
        {
            dp[i][0] = dp[i-1][0];
            dp[i][1] = max(dp[i-1][0] - prices[i] , dp[i-1][1]);
            dp[i][2] = max(dp[i-1][1] + prices[i] , dp[i-1][2]);
            dp[i][3] = max(dp[i-1][2] - prices[i] , dp[i-1][3]);
            dp[i][4] = max(dp[i-1][3] + prices[i] , dp[i-1][4]);
        }
        return max(dp[prices.size()-1][2] , dp[prices.size()-1][4] );
    }
};