122.买卖股票的最佳时机Ⅱ
给你一个整数数组prices,其中*prices[i]*表示某支股票第 i 天的价格。
在每一天,你可以决定是否购买和/或出售股票。你在任何时候最多只能持有一股股票。你也可以先购买,然后在同一天出售。
返回:你能获得的最大利润 。
题解
class Solution:
def maxProfit(self, prices: List[int]) -> int:
l = len(prices)
if l < 2:
return 0
dp = [[0, 0] for i in range(l)]
dp[0][1] = - prices[0]
for i in range(1, l):
dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] + prices[i])
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] - prices[i])
return dp[l - 1][0]
思路
使用动态规划求解。首先,这题用贪心更简单。
首先确定状态表示,然后找到状态转移方程。
状态表示:每一天的结束时会有两种状态:持有股票、未持有股票;那么使用一个i * 2的矩阵dp[i][2]表示第i天结束未持有股票(dp[i][0])和持有股票(dp[i][1])的收益。
状态转移方程:后一天赚的钱和前一天是相关的。
第i天结束未持有股票有两种情况:第i - 1天也没股票、第i - 1天有股票但第i天卖了。
第i天结束未持有股票收益:dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] + prices[i])
第i天结束持有股票有两种情况:第i - 1天没股票第i天买入了股票、第i - 1天有股票。
第i天结束持有股票收益:dp[i][0] = max(dp[i - 1][0] - prices[i], dp[i - 1][1])
然后给第一天的数值进行初始化,遍历一遍就得到结果了。