分類
- 題目
- 解題思路
- Python實現
題目
給定一個數組,它的第 i 個元素是一支給定的股票在第 i 天的價格。
設計一個算法來計算你所能獲取的最大利潤。你最多可以完成 兩筆 交易。
注意: 你不能同時參與多筆交易(你必須在再次購買前出售掉之前的股票),地址。
示例:
輸入: [3,3,5,0,0,3,1,4]
輸出: 6
解釋: 在第 4 天(股票價格 = 0)的時候買入,在第 6 天(股票價格 = 3)的時候賣出,這筆交易所能獲得利潤 = 3-0 = 3 。
隨后,在第 7 天(股票價格 = 1)的時候買入,在第 8 天 (股票價格 = 4)的時候賣出,這筆交易所能獲得利潤 = 4-1 = 3 。
解題思路
最佳買賣股票時機含冷凍期
類似于含冷凍期的買賣股票,也是一個狀態機。本次買賣股票共有四種狀態:第一次買賣股票,第一次賣出股票后,第二次買入股票,第二次賣出股票后分別設位S1,S2,S3,S4,并且對應的狀態轉移方程:
s1
=
max
(
s1
,
-
prices
[
i
]
)
s2
=
max
(
s2
,
s1
+
prices
[
i
]
)
s3
=
max
(
s3
,
s2
-
prices
[
i
]
)
s4
=
max
(
s4
,
s3
+
prices
[
i
]
)
Python實現
class
Solution
:
def
maxProfit
(
self
,
prices
:
List
[
int
]
)
-
>
int
:
size
=
len
(
prices
)
if
size
==
0
:
return
0
# s1是假設第一天就買入股票,s2,s3,s4則分別設為最小值
s1
=
-
prices
[
0
]
s2
,
s3
,
s4
=
float
(
'-inf'
)
,
float
(
'-inf'
)
,
float
(
'-inf'
)
for
i
in
range
(
size
)
:
s1
=
max
(
s1
,
-
prices
[
i
]
)
s2
=
max
(
s2
,
s1
+
prices
[
i
]
)
s3
=
max
(
s3
,
s2
-
prices
[
i
]
)
s4
=
max
(
s4
,
s3
+
prices
[
i
]
)
return
max
(
0
,
s4
)
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
