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

【LeetCode】Best Time to Buy and Sell Stock

时间:2014-05-12 05:56:14      阅读:315      评论:0      收藏:0      [点我收藏+]

标签:des   style   blog   class   code   java   

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

标签:des   style   blog   class   code   java   

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

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