dp定义为dp[i][k][0]第i天没有持股,恰好完成了k笔交易(卖出时算一笔交易完成)的情况下可以获得的最大利润
dp[0][k>=1][0or1] 不合法,因为是第一天不可能进行过交易
这题就是k=2
注意按照这个通解,由于定义的是恰好完成k笔交易在卖出时算交易完成,根据递推公式无法涉dp[i][0][1]
所以这一步放在k的循环之外
def maxProfit(self, k, prices):
"""
:type k: int
:type prices: List[int]
:rtype: int
"""
#dp定义为dp[i][k][0]第i天没有持股,恰好完成了k笔交易(卖出时算一笔交易完成)的情况下可以获得的最大利润
#dp[0][k>=1][0or1] 不合法,因为是第一天不可能进行过交易
n = len(prices)
k = min(k,n//2)
res = 0
dp = [[[0]*2 for j in range(k+1)] for i in range(n)]
dp[0][0][1] = - prices[0]
for j in range(1,k):
dp[0][j][0] = -0x3f3f3f3f
dp[0][j][1] = -0x3f3f3f3f
for i in range(1,n):
dp[i][0][1] = max(dp[i - 1][0][1], dp[i - 1][0][0] - prices[i]) #第i天进行了0次交易持有股票
for j in range(k,0,-1): #持有股票在0次交易的时候没有考虑到
dp[i][j][0] = max(dp[i-1][j][0],dp[i-1][j-1][1] + prices[i])
dp[i][j][1] = max(dp[i-1][j][1],dp[i-1][j][0] - prices[i])
for i in range(1,k+1):
res = max(res,dp[n-1][i][0])
return res
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
n = len(prices)
dp = [[0]*4 for i in range(n)] #四个状态分别代表第一次买入,第一次卖出,第二次买入,第二次卖出
dp[0][0] = -prices[0]
dp[0][2] = -prices[0] #注意这里
for i in range(1,n):
dp[i][0] = max(dp[i-1][0],-prices[i])
dp[i][1] = max(dp[i-1][1],dp[i-1][0]+prices[i])
dp[i][2] = max(dp[i-1][2],dp[i-1][1]-prices[i])
dp[i][3] = max(dp[i-1][3],dp[i-1][2]+prices[i])
return max(dp[n-1][1],dp[n-1][3])