题目描述
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例1
输入: [2,4,1], k = 2 输出: 2 解释: 在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例2
输入: [3,2,6,5,0,3], k = 2 输出: 7 解释 :在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
题解
这是 【买卖股票的最佳时机】 系列题目的第四题。
代码
python
classSolution: defmaxProfit(self, k: int, prices: List[int]) ->int: n=len(prices) ifn==0: return0ifk>=n//2: res=0foriinrange(1, n): res+=max(prices[i]-prices[i-1], 0) returnresdp0= [-prices[0]] * (k+1) dp1= [0] * (k+1) forpinprices[1:]: foriinrange(1, k+1): dp1[i] =max(dp1[i], dp0[i]+p) dp0[i] =max(dp0[i], dp1[i-1]-p) returnmax(dp1[k], 0)
作者简介:godweiyang,知乎同名,华东师范大学计算机系硕士在读,方向自然语言处理与深度学习。喜欢与人分享技术与知识,期待与你的进一步交流~