题目描述:
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-ii
动态规划,解题思路详见代码注释。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
n = len(prices)
if n == 0:
return 0 #么得股票买卖
dp0 = 0 #第一天得状态是卖股票,那么此时的利润为0
dp1 = -prices[0] #第一天的状态是买股票,那么此时的利润为-prices[0]
for i in range(1,n):
'''
注释1:
题目要求手中只能有一只股票
1.前一天是卖股票的状态时,第二天有两种选择:a.买股票,处在买股票的状态(状态转移到2);
b.依旧处在卖股票的状态,即不操作
2.前一天是买股票的状态时,第二天有两种选择:a.卖股票,处在卖股票的状态(状态转移到1);
b.依旧处在买股票的状态,即不操作
'''
dp0 = max(dp0, dp1 + prices[i]) #与注释1相对的,要处在卖股票的状态,前一天要是买,当天卖,或者当天就不进行操作
dp1 = max(dp1, dp0 - prices[i] ) # 买与注释1相对的,要处在买股票的状态,前一天要是卖,当天买,或者当天就不进行操作
return max(dp0,dp0)