题目: 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 as many transactions as you like (i.e., buy one and sell one share of the stock multiple times). Note: You may not engage in multiple transactions at the same time (i.e., you must sell the stock before you buy again).
理解:
假设你有个数组,其中第i个元素是第i天的股票价格,请设计一种算法去找到最大的收益,你可以用任何你喜欢的方式完成交易(多次买入或卖出股票权)。
注意:你不能在同一时间完成多次交易,你必须再次买入这支股票以后才能卖出。
例子1:
Input: [7,1,5,3,6,4] Output: 7 Explanation: Buy on day 2 (price = 1) and sell on day 3 (price = 5), profit = 5-1 = 4. Then buy on day 4 (price = 3) and sell on day 5 (price = 6), profit = 6-3 = 3.
例子2:
Input: [1,2,3,4,5] Output: 4 Explanation: Buy on day 1 (price = 1) and sell on day 5 (price = 5), profit = 5-1 = 4. Note that you cannot buy on day 1, buy on day 2 and sell them later, as you are engaging multiple transactions at the same time. You must sell before buying again.
例子3:
Input: [7,6,4,3,1] Output: 0 Explanation: In this case, no transaction is done, i.e. max profit = 0.
原始解题思路:
要获得最好的收益,就要坚持“低吸高抛”,所以只需要比较当前日的股价与下一日的股价,如果下一日的股价高于本日,那就购买,计算收益(此时为负数),否则继续等待。
第二天继续比较第二天与第三天的股价,如果第三天高于第二天,则购买(此时相当于持有不做出操作,设置一个标志符表示当前是已买入还是已卖出),如果第三天低于第二天的股价,则卖出。
python代码:
class Solution: def maxProfit(self, prices): trans_flag = 0 # 0 表示未购买或者已卖出,1 表示已购买未卖出 profit = 0 end_day = len(prices) if end_day < 2: print("总收益:{}".format(profit)) return 0 for i in range(end_day - 1): if prices[i] < prices[i + 1] and trans_flag == 0: print("买入第{}天的股票".format(i + 1)) trans_flag = 1 current_i = i profit -= prices[i] continue if prices[i] < prices[i + 1] and trans_flag == 1: print("继续持有股票") continue if prices[i] > prices[i + 1] and trans_flag == 1: print("卖出第{}天的股票".format(i + 1)) profit += prices[i] current_i = 0 trans_flag = 0 continue if prices[i] > prices[i + 1] and trans_flag == 0: print("暂不购买") continue print("当前收益:{}".format(profit)) # 第一个问题:这里加一步验证一直持有到最后一天未卖出,所以收益没计算完全的问题,同时要小心很小的数据集,比如只有一个 # 第二个问题:与上一天比较时不一定要大于,可能两者相等 if prices[end_day - 1] >= prices[end_day - 2] and trans_flag == 1: trans_flag = 1 profit += prices[end_day - 1] print("总收益:{}".format(profit)) if __name__ == '__main__': prices1 = [7, 1, 5, 3, 6, 4] prices2 = [1, 2, 3, 4, 5] prices3 = [7, 6, 4, 3, 1] prices4 = [2,3] prices5 = [1,9,6,9,1,7,1,1,5,9,9,9] Mine = Solution() Mine.maxProfit(prices5)
验证结果:
Runtime:
44 ms, faster than 67.87% of Python3 online submissions for Best Time to Buy and Sell Stock II.
Memory Usage:
13.8 MB, less than 5.06% of Python3 online submissions for Best Time to Buy and Sell Stock II.
改进版本1:
这个结果还是比较满意的,但是仍有优化的空间,首先是判断if太多,因为我们只需要比较当前一天与下一天的大小,如果下一天减去当前天的股票价格的结果大于0那么就可以交易,
因此实际上一个for循环就可以搞定,不去管是否已经交易了,而是直接将两者做差作为交易结果,只要结果大于等于0就累加。其实就省去了第一个思路的最后一步,将前面的大于改为了大于等于就可以交易(反正结果是0)
python代码:
class Solution: def maxProfit2(self, prices): print(sum(max(prices[i + 1] - prices[i], 0) for i in range(len(prices) - 1)))
验证结果:
Runtime:
44 ms, faster than 67.87% of Python3 online submissions for Best Time to Buy and Sell Stock II.
Memory Usage:
13.9 MB, less than 5.06% of Python3 online submissions for Best Time to Buy and Sell Stock II.
改进版本2:
可以看到这样写大大简化了代码量,但是性能方面并没有提升多少,那么还可以进一步优化吗?那就是使用python迭代器来替换list数组的range,下面简要搬运一下python的zip函数教程:
zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的对象,这样做的好处是节约了不少的内存。
我们可以使用 list() 转换来输出列表。如果各个迭代器的元素个数不一致,则返回列表长度与最短的对象相同,利用 * 号操作符,可以将元组解压为列表。
>>>a = [1,2,3] >>> b = [4,5,6] >>> c = [4,5,6,7,8] >>> zipped = zip(a,b) # 返回一个对象 >>> zipped>>> list(zipped) # list() 转换为列表 [(1, 4), (2, 5), (3, 6)] >>> list(zip(a,c)) # 元素个数与最短的列表一致 [(1, 4), (2, 5), (3, 6)] >>> a1, a2 = zip(*zip(a,b)) # 与 zip 相反,zip(*) 可理解为解压,返回二维矩阵式 >>> list(a1) [1, 2, 3] >>> list(a2) [4, 5, 6] >>>
在这里就是将range函数变为迭代器,每次从prices的当前位置和下一个位置取出x和y,然后依然判断求和
python代码:
class Solution: def maxProfit(self, prices): return sum([y - x for x, y in zip(prices[:-1], prices[1:]) if x < y])
验证结果:
Runtime:
40 ms, faster than 99.02% of Python3 online submissions for Best Time to Buy and Sell Stock II.
Memory Usage:
14 MB, less than 5.06% of Python3 online submissions for Best Time to Buy and Sell Stock II.
我在这题又发现了一个有趣的结果,那就是如果将解法三的if换成解法二的max函数,那么速度就差不多了,但是如果将解法二的max函数去掉改为if,速度并没有得到加快,
因此只要采用python list寻找特定位置的方式,速度就会有所降低,而即便是采用迭代器方法,加入max函数某些程度上也会降低速度。
附上解法二用if改掉max函数的python代码
class Solution: def maxProfit(self, prices): return(sum(prices[i + 1] - prices[i] for i in range(len(prices) - 1) if prices[i + 1] > prices[i]))
结果:
Runtime:
68 ms, faster than 5.85% of Python3 online submissions for Best Time to Buy and Sell Stock II.
Memory Usage:
14.2 MB, less than 5.06% of Python3 online submissions for Best Time to Buy and Sell Stock II.
所以想提高,还是要学一下迭代器的应用啊。