本篇内容主要讲解“如何用leetcode计算出买卖股票的最佳时机”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“如何用leetcode计算出买卖股票的最佳时机”吧!
一、题目内容
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入:k = 2, prices = [2,4,1]
输出:2
解释:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例 2:
输入:k = 2, prices = [3,2,6,5,0,3]
输出:7
解释:在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
提示:
0 <= k <= 109
0 <= prices.length <= 1000
0 <= prices[i] <= 1000
二、解题思路
动态规划,每个price都需要检查一遍;
然后买入和卖出的两个状态分别进行max比较即可。
三、代码
import sys class Solution: def maxProfit(self, k: int, prices: list) -> int: dp = [[0, 0] for _ in range(k + 1)] for j in range(k + 1): dp[j][1] = -sys.maxsize for i in range(0, len(prices)): for j in range(1, k + 1): dp[j][0] = max(dp[j][0], dp[j][1] + prices[i]) # 卖出 dp[j][1] = max(dp[j][1], dp[j - 1][0] - prices[i]) # 买入 return dp[k][0] if __name__ == '__main__': k = 2 prices = [3, 2, 6, 5, 0, 3] s = Solution() ans = s.maxProfit(k, prices) print(ans)
到此,相信大家对“如何用leetcode计算出买卖股票的最佳时机”有了更深的了解,不妨来实际操作一番吧!这里是亿速云网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!
原创文章,作者:奋斗,如若转载,请注明出处:https://blog.ytso.com/223614.html