121. Best Time to Buy and Sell Stock (python)

来源:互联网 发布:java语言程序设计下载 编辑:程序博客网 时间:2024/05/17 02:26

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.
Example 1:
Input: [7, 1, 5, 3, 6, 4]
Output: 5
max. difference = 6-1 = 5 (not 7-1 = 6, as selling price needs to be larger than buying price)
Example 2:
Input: [7, 6, 4, 3, 1]
Output: 0
In this case, no transaction is done, i.e. max profit = 0.
题意:数组第i个元素是第i天股票的销售价格,记住只允许进行一次交易,只允许买一支股票并卖掉,求最大的收益。
思路:动态规划法。
从前向后遍历数组,记录当前出现过的最低价格,作为买入价格,并计算以当天价格出售的收益,作为可能的最大收益,整个遍历过程中,出现过的最大收益就是所求。
复杂度:O(n)时间,O(1)空间。
参考:http://liangjiabin.com/blog/2015/04/leetcode-best-time-to-buy-and-sell-stock.html

class Solution(object):    def maxProfit(self, prices):        if len(prices)<2:            return  0        maxpf=0        low=prices[0]        for i in range(len(prices)):            if prices[i]<low:                low=prices[i]            maxpf=max(prices[i]-low,maxpf)        return maxpf
0 0
原创粉丝点击