首页 > 其他 > 详细

【LeetCode】Best Time to Buy and Sell Stock

时间:2014-05-12 05:53:40      阅读:374      评论:0      收藏:0      [点我收藏+]

Best Time to Buy and Sell Stock

 

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

If you were only permitted to complete at most one transaction (ie, buy one and sell one share of the stock), design an algorithm to find the maximum profit.

 

这题不难,第一反应就是O(n)解决。

关键点就在于,随着prices的遍历,怎样更新gap?

我的做法是,

如果当前遇到的元素比当前max大,就更新max并更新gap

如果当前遇到的元素比当前min小,就重新进行gap的计算

成功AC,上代码

bubuko.com,布布扣
class Solution {
public:
    int maxProfit(vector<int> &prices) 
    {
        if(prices.empty())
            return 0;
        
        int maxGap = 0;
        int curGap = 0;
        int max = prices[0];
        int min = prices[0];
        curGap = max - min;
        
        for(vector<int>::size_type st = 0; st < prices.size(); st ++)
        {
            if(prices[st] > max)
            {
                max = prices[st];
                curGap = max - min;
                if(curGap > maxGap)
                    maxGap = curGap;
            }
            else if(prices[st] < min)
            {
                min = prices[st];
                max = prices[st];
                curGap = max - min;
            }
        }
        return maxGap;
    }
};
bubuko.com,布布扣

bubuko.com,布布扣

 

后来看到Discussion中一种更简洁的做法,

关键点是,如果最大gap对应的max为第j个元素,那么对应的min必是prices[j]之前的最小元素

bubuko.com,布布扣
class Solution 
{
public:
    int maxProfit(vector<int> &prices) 
    {
        if(prices.empty())
            return 0;
        
        int maxGap = 0;
        int curGap = 0;
        int min = prices[0];
        
        for(vector<int>::size_type st = 1; st < prices.size(); st ++)
        {
            if(prices[st] < min)
                min = prices[st];
            
            else
            {
                curGap = prices[st] - min; 
                if(curGap > maxGap)
                    maxGap = curGap;
            }
            
        }
        
        return maxGap;
    }
};
bubuko.com,布布扣

bubuko.com,布布扣

【LeetCode】Best Time to Buy and Sell Stock,布布扣,bubuko.com

【LeetCode】Best Time to Buy and Sell Stock

原文:http://www.cnblogs.com/ganganloveu/p/3721162.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!