算法学习打卡计划

leetcode第一百二十一题——买卖股票的最佳时机

2020-02-16  本文已影响0人  不分享的知识毫无意义

这道题是面试遇到的,显然是个动态规划题。

1.题目

原题:

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。注意你不能在买入股票前卖出股票。

例子:

输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。 注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。

2.解析

(1)暴力求解法
这道题的解法最好想的就是暴力求解,复杂度O(n2),但是要注意买卖只能从买入之后进行所以第二个for循环写i+1就行了。
(2)动态规划法
但是,肯定有更优的方法,比如动态规划就是为了这道题而存在的,一开始我想的是用一个二维列表来做,但是仔细一想,他的复杂度跟暴力差不多啊。后来借鉴网上思路了,用一个一维列表存储截止第k天的最大收益,还有一个变量记录截止昨天的最低买入价格。这样一来时间复杂度为O(n)了。因为是动态规划,所以最大值肯定是最后一个元素,这样就省去排序这一步了。

3.python代码

class Solution:
    def  maxProfit(self, prices):
        if not prices or len(set(prices)) == 1:
            return 0
        m = len(prices)
        maxout = [0]
        minp = prices[0]
        for i in range(1, m):
            maxout.append(max(prices[i] - minp, maxout[i-1]))
            if prices[i] < minp:
                minp = prices[i]
        return maxout[-1]
上一篇下一篇

猜你喜欢

热点阅读