码迷,mamicode.com
首页 > 其他好文 > 详细

[动态规划] leetcode 309 Best Time to Buy and Sell Stock with Cooldown

时间:2019-08-04 19:45:14      阅读:81      评论:0      收藏:0      [点我收藏+]

标签:记录   动态   没有   tor   https   com   cto   tps   vector   

problem:https://leetcode.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/

        这道题有好几个变种。对于没有cooldown的题目而言,只需要考虑第i天,当前持有股票的最大收益和当前不持有股票的最大收益。而引入cooldown后,我们也需要引入新的状态:

        hasStock : 记录当天持有股票时的最大收益:可能是之前就有的,也可能是当天买的(由于必须是冷却一天才能买新的股票,需要从冷却状态转换过来),取两者的最大值。

        cool:记录当天的操作选择冷却时的最大收益,可以选择一直冷却,也可以刚卖完股票进入冷却;取两者的最大值。

        sell:记录当天选择出售股票后的收益。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        
        int n = prices.size();
        vector<int> hasStock(n + 1, -500000);
        vector<int> sell(n + 1, -500000);
        vector<int> cool(n + 1, 0);
        int res = 0;
        for(int i = 0;i < n;i++)
        {
            hasStock[i + 1] = max(hasStock[i], cool[i] - prices[i]);
            cool[i + 1] = max(sell[i], cool[i]);
            sell[i + 1] = hasStock[i] + prices[i];
        }
        return max(cool[n], sell[n]);
    }
};

 

[动态规划] leetcode 309 Best Time to Buy and Sell Stock with Cooldown

标签:记录   动态   没有   tor   https   com   cto   tps   vector   

原文地址:https://www.cnblogs.com/fish1996/p/11299262.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!