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

LeetCode——Best Time to Buy and Sell Stock III (股票买卖时机问题3)

时间:2014-10-12 00:58:36      阅读:686      评论:0      收藏:0      [点我收藏+]

标签:des   style   blog   http   color   io   os   ar   java   

问题:

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.

依旧还是那个熟悉的味道,如果III的另一个版本,那么III就是他们的一个升级版。

在这个版本中只能交易两次,也就是买卖各两次,其它条件依旧没有变(不熟悉的请参见I

分析:

 

在此题中,最多只能买卖两次,而且两次交易不能有交叉,也就是说应该是这样儿的(直线为时间,弧线为交易),

bubuko.com,布布扣  或者  bubuko.com,布布扣

不会出现:

bubuko.com,布布扣

问题三是问交易两次的最大收益,从上面的图我们可以看出由于两次交易没有交叉,我们可以考虑把这个问题拆分为两个问题:

1.第一次交易的最大收益(在前)

2.第二次交易的最大收益(在后)

那么现在问题又来了,在哪个位置拆分呢?几经思考发现从哪儿拆分都不合适,无论从哪儿拆分都不敢保证这种拆分办法就能得出最大收益,既然哪个位置拆分都不能保证最大收益,那如果我们每个位置都拆分一次,然后比较各自的最大收益,这样是否可行呢?在股票买卖时机问题1中我们用时间复杂度O(n)解决了一次交易的最大收益问题,那么在这里,每一次拆分会产生两个组,每个组用问题1的解法解答,那每一次拆分找出最大收益的时间复杂度为O(n),那拆分n-1次的时间复杂度就是O(n^2)了。按这种方法可以解决了,但是在LeetCode上提交貌似会报超时,而且,问题I和II都是在时间复杂度O(1)内解决的,没道理问题III要O(n^2),我们再仔细的考虑一下……

 

在问题I中我们采用动态规划的思想,通过分阶段找最优策略,后一阶段的最优策略由前一阶段的最优策略转化而来,换个角度,我们可以认为我们从第一阶段起,我们找到了每个阶段的最优解,在这道题中,也就是我们找到了从第1天到第i天的最大收益(i=2,3,4,…,n),只是在问题I中我们并没有把这些各个阶段的最大收益值保存下来。

 

那么我们回到问题III,我们先用I的办法把数组遍历一遍,并且声明一个数组profit[n],profit[i]保存第一天到第i+1天的最大收益,i=0,1,2,…,n-1,到此时间复杂度为O(n)。

此时相当于我们知道拆分后(假设从数组下标k处拆分)的第一组的最大收益,那么第二组呢?profit[i]只是记录了从0到i的最大收益,我们如何知道从k到i的最大收益呢?

 

再次分析问题I的解法,问题I的重点代码如下(完整的请参考http://www.cnblogs.com/chrischennx/p/4014422.html):

for(int i=1; i<prices.length;i++){
           min = prices[i]<min?prices[i]:min;
           profit = prices[i]-min>profit?prices[i]-min:profit;
        }

它是通过找到各个阶段的最小值,然后用当前值减去最小值和前一阶段的最大收益比较得到最大收益,那么在问题III拆分后的第二组中我们是否也可以考虑用类似的办法来处理呢,每次拆分再计算k到i的最大收益肯定是不行的,因为每次时间复杂度为O(n),拆了n-1次时间复杂度就成了O(n^2),又回到刚开始的了,这不是我们需要的。

在问题I中是按数组从前往后考虑的,那么在第二组中,由于每次拆分都有最后一个值a[n-1],那么索性我们第二组从最后开始考虑。问题I是考虑前边儿找一个低的价格买入,后面找个高价卖出,也就是先找买入的价格,再找卖出的价格;那么从后开始考虑可以理解为:后面找个高价卖出,前面找个低价买入,也就是先找卖出的价格,再找买入的价格。也就是这样:

for(int i=a.length-2;i>=0;i--){
            max = a[i]<max?max:a[i];
            profit2[i] = max-a[i]>profit2[i+1]?max-a[i]:profit2[i+1];
        }

然后我们就可以通过profit[i]+profit2[i]的最大值得到两次交易的最大收益。

 

到此问题就可以解决了,但是此时的空间复杂度为O(2n),其实还有更省的:不求profit2数组。

我们如此这般:

int max = a[a.length-1];
        int result = profit[a.length-1];
        for(int i=a.length-2;i>=0;i--){
            max = a[i]>max?a[i]:max;
            result = profit[i]+max-a[i]>result?profit[i]+max-a[i]:result;
        }

其中profit[i]是按问题I的方法把求出的0到i的最大收益。

result = profit[i]+max-a[i]>result?profit[i]+max-a[i]:result;

这里用result直接用来保存最大收益,乍一看,容易理解为profit[i]+(max-a[i])和result比较,但是max-a[i]并不一定是第二组的最大收益,比如:[5,6,1,4],当i=1的时候,a[i]=6,max=6,但是此时的profit[1]+(max-a[1]),也就是profit[1]+0,肯定没有i=2,a[i]=1,max=4的时候大,profit[2]+3,看似这么理解是对的,但是,其实是错的,因为profit[i]+max-a[i]并不是在和它自己比,而是和result比,若之前的比较中有比较大的,那么就保存到result中了,之后的值比result小,result的值就不变了,所以最终result还是最大值。

如:[5,6,1,4]这个例子,

用问题I的方法计算出profit=[0,1,1,3],

当i=2的时候result=4,

当i=1的时候result = 1+4-4>4?1+4-4:4 = 4。

 

 


 

完整代码如下(java):

public class Solution {
    public int maxProfit(int[] a) {
        if(a==null || a.length==0)return 0;
        int profit[] = new int[a.length];
        int min = a[0];
        profit[0] = 0;
        for(int i=1;i<a.length;i++){
            min = a[i]<min?a[i]:min;
            profit[i] = a[i]-min>profit[i-1]?a[i]-min:profit[i-1];
        }
        
        int max = a[a.length-1];
        int result = profit[a.length-1];
        for(int i=a.length-2;i>=0;i--){
            max = a[i]>max?a[i]:max;
            result = profit[i]+max-a[i]>result?profit[i]+max-a[i]:result;
        }
        
        return result;
    }
}

LeetCode——Best Time to Buy and Sell Stock III (股票买卖时机问题3)

标签:des   style   blog   http   color   io   os   ar   java   

原文地址:http://www.cnblogs.com/chrischennx/p/4019905.html

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