[leetcode][DP] Best Time to Buy and Sell Stock III

题目:

Say you have an array for which the ith element is the price of a given stock on day i.

Design an algorithm to find the maximum profit. You may complete at most two transactions.

Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n <= 1) return 0;
        int lowestPrice = prices[0];//从第0开始到当天的最低价
        int highestPrice = prices[n-1];//从当天到最后一天的最高价
        //创建table
        int *tableForward = new int[n];//tableForward[i]表示0...i天内所得的最大利润
        int *tableBackward = new int[n];//tableBackward[i]表示i...n天内所得的最大利润
        //table初始化
        tableForward[0] = 0;
        tableBackward[n-1] = 0;
        //填tableForward 
        for(int i = 1; i < n; ++i){
            if(prices[i] < lowestPrice) lowestPrice = prices[i];
            tableForward[i] = prices[i]-lowestPrice > tableForward[i-1] ? prices[i]-lowestPrice : tableForward[i-1];
        }
        //填tableBackward
        for(int i = n-2; i >= 0; --i){
            if(prices[i] > highestPrice) highestPrice = prices[i];
            tableBackward[i] = highestPrice - prices[i] > tableBackward[i+1] ?  highestPrice - prices[i] : tableBackward[i+1];
        }
        //选出对应位置之和的最大值
        int res = tableForward[0] + tableBackward[0];
        for(int i = 1; i < n; ++i){
            if(tableForward[i]+tableBackward[i] > res) res = tableForward[i]+tableBackward[i];
        }
        //释放内存
        delete []tableForward;
        delete []tableBackward;
        //返回结果
        return res;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值