Leetcode刷题笔记 122. 买卖股票的最佳时机 II

时间：2020年11月7日
知识点：动态规划
题目链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-ii//

题目
给定一个数组，它的第 i 个元素是一支给定股票第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易（多次买卖一支股票）。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1:
输入:
[7,1,5,3,6,4]
输出:
7
解释:
在第 2 天（股票价格 = 1）的时候买入，在第 3 天（股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
随后，在第 4 天（股票价格 = 3）的时候买入，在第 5 天（股票价格 = 6）的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。

示例2 :

输入: [1,2,3,4,5]
输出: 4
解释:
在第 1 天（股票价格 = 1）的时候买入，在第 5 天 （股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接连购买股票，之后再将它们卖出。
因为这样属于同时参与了多笔交易，你必须在再次购买前出售掉之前的股票。

示例 3:
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

提示：

• 1 <= prices.length <= 3 * 10 ^ 4
• 0 <= prices[i] <= 10 ^ 4

解法：

1. 一看就是动态规划 一般来说动态规划的第一个是变化的天数 第二个是状态
2. 这里有两个状态： 手上有股票 手上没股票 时能获得的最大利润
3. 可以得出转移方程
   1. 当天手上没股票 = max（昨天手上没股票，昨天手上有股票+今天股票价格
   2. 当天手上有股票 = max（昨天手上有股票，昨天手上没股票-今天股票价格）
4. 返回最后一天手上没股票的
5. 状态压缩

代码

#include <stdio.h>
#include <vector>
#include <iostream>
using namespace std;
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        int dp[n][2];
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for(int i = 1;i < n;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[n-1][0];
    }
};
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        int dp0 = 0, dp1 = -prices[0];
        for (int i = 1; i < n; i++) {
            int d0 = max(dp0, dp1 + prices[i]);
            int d1 = max(dp1, dp0 - prices[i]);
            dp0 = d0;
            dp1 = d1;
        }
        return dp0;
    }
};

int main()
{
    vector<int> prices{7,1,5,3,6,4};
    Solution s;
    cout<<s.maxProfit(prices)<<endl;
    return 0;
}

今天也是爱zz的一天哦！