leetcode笔记(一)309. Best Time to Buy and S…

2018-06-17 20:59:28来源:未知 阅读 ()

新老客户大回馈,云服务器低至5折

  • 题目描述 (原题目链接)

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 as many transactions as you like (ie, buy one and sell one share of the stock multiple times) with the following restrictions:

  • You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
  • After you sell your stock, you cannot buy stock on next day. (ie, cooldown 1 day)

Example:

prices = [1, 2, 3, 0, 2]
maxProfit = 3
transactions = [buy, sell, cooldown, buy, sell]
  • 解题思路 (原思路链接)
这道题目自己一开始根本没有想到用动态规划,菜鸟本色。之后看了dicussion区域排名第一的答案,也是云里雾里。
还好碰到了这个解决思路。这个方案把每天分为四个状态
  1. 有股票,卖 = 前天(有股票,休息)+ price 或者 前天(没股票,买)+ price
  2. 有股票,休息 = 前天(有股票,休息)或者 前天(没股票,买)
  3. 没股票,买 = 前天(没股票,休息)- price ==》 冷却期所以不可能是 有股票,卖
  4. 没股票,休息 = 前天(没股票,休息)或者 前天(有股票,卖)

结合这个思路,编码实现如下:

需要额外注意的一点是第0天的初始化,(有股票,休息)= -price 因为下一天可能卖这个股票,所以计价时相当于第0天买了。

    int maxProfit(vector<int>& prices) {
        
        int has_sell, has_sell_before;
        int has_rest, has_rest_before;
        int no_buy, no_buy_before;
        int no_rest, no_rest_before;
        
        int size = prices.size();
        if(size < 2)
            return 0;
        has_sell_before = 0;
        has_rest_before = -prices[0];
        no_buy_before = -prices[0];
        no_rest_before = 0;
        for(int i = 1; i < size; i++)
        {
            has_sell = max(has_rest_before + prices[i], no_buy_before + prices[i]);
            has_rest = max(has_rest_before, no_buy_before);
            no_buy = no_rest_before - prices[i];
            no_rest = max(no_rest_before, has_sell_before);
            
            has_sell_before = has_sell;
            has_rest_before = has_rest;
            no_buy_before = no_buy;
            no_rest_before = no_rest;
        }
        
        // find the max between has_sell and no_rest
        return max(has_sell, no_rest);
    }

 

标签:

版权申明:本站文章部分自网络,如有侵权,请联系:west999com@outlook.com
特别注意:本站所有转载文章言论不代表本站观点,本站所提供的摄影照片,插画,设计作品,如需使用,请与原作者联系,版权归原作者所有

上一篇:PTA练习题之7-1 出租车计价(15 分)

下一篇:洛谷P1962 斐波那契数列(矩阵快速幂)