9.23 动态规划例题

买卖股票的最好时机(三)_牛客题霸_牛客网 (nowcoder.com)

BM82 买卖股票的最好时机(三)

描述

假设你有一个数组prices，长度为n，其中prices[i]是某只股票在第i天的价格，请根据这个价格数组，返回买卖股票能获得的最大收益
1. 你最多可以对该股票有两笔交易操作，一笔交易代表着一次买入与一次卖出，但是再次购买前必须卖出之前的股票
2. 如果不能获取收益，请返回0
3. 假设买入卖出均无手续费

数据范围：1 \le n \le 10^51≤n≤105，股票的价格满足 1 \le val\le 10^41≤val≤104

要求: 空间复杂度 O(n)，时间复杂度 O(n)

进阶：空间复杂度 O(1)，时间复杂度 O(n)

示例1

输入：[8,9,3,5,1,3]
返回值：4
说明：第三天(股票价格=3)买进，第四天(股票价格=5)卖出，收益为2 第五天(股票价格=1)买进，第六天(股票价格=3)卖出，收益为2 总收益为4。

示例3

输入：[1,2,8,3,8]

返回值：12

说明：第一笔股票交易在第一天买进，第三天卖出；第二笔股票交易在第四天买进，第五天卖出；总收益为12。 因最多只可以同时持有一只股票，所以不能在第一天进行第一笔股票交易的买进操作，又在第二天进行第二笔股票交易的买进操作（此时第一笔股票交易还没卖出），最后两笔股票交易同时在第三天卖出，也即以上操作不满足题目要求。

思路：对于此类求极值且情况多，前面的选择会影响后面的结果和选择时，考虑使用贪心和动态规划

初始状态：f[0][0]=0(第一天不买)

                  f[0][3]=-prices[0](第一天购买)

递推方式：

class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 两次交易所能获得的最大收益
     * @param prices int整型vector 股票每一天的价格
     * @return int整型
     */
    int maxProfit(vector<int>& prices) {
        // write code here
        vector<vector<int>> f(prices.size(), vector<int>(5, -10000));
        f[0][0] = 0;
        f[0][3] = -1 * prices[0];
        for (int i = 1; i < prices.size(); i++) {
            f[i][0] = f[i-1][0];
            f[i][1] = max(f[i - 1][1], f[i - 1][3] + prices[i]);
            f[i][2] = max(f[i - 1][2], f[i - 1][4] + prices[i]);
            f[i][3] = max(f[i - 1][3], f[i - 1][0] - prices[i]);
            f[i][4] = max(f[i - 1][4], f[i - 1][1] - prices[i]);
        }
        //最后不持有股票
        return max(max(0,f[prices.size() - 1][2]), f[prices.size() - 1][1]);
    }
};