leetcode:买卖股票的最佳时机2(python)
1. 題目描述
給定一個(gè)數(shù)組,它的第 i 個(gè)元素是一支給定股票第 i 天的價(jià)格。
設(shè)計(jì)一個(gè)算法來計(jì)算你所能獲取的最大利潤。你可以盡可能地完成更多的交易(多次買賣一支股票)。
注意:你不能同時(shí)參與多筆交易(你必須在再次購買前出售掉之前的股票)。
示例 1:
輸入: [7,1,5,3,6,4]
輸出: 7
解釋: 在第 2 天(股票價(jià)格 = 1)的時(shí)候買入,在第 3 天(股票價(jià)格 = 5)的時(shí)候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
隨后,在第 4 天(股票價(jià)格 = 3)的時(shí)候買入,在第 5 天(股票價(jià)格 = 6)的時(shí)候賣出, 這筆交易所能獲得利潤 = 6-3 = 3 。
示例 2:
輸入: [1,2,3,4,5]
輸出: 4
解釋: 在第 1 天(股票價(jià)格 = 1)的時(shí)候買入,在第 5 天 (股票價(jià)格 = 5)的時(shí)候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接連購買股票,之后再將它們賣出。
因?yàn)檫@樣屬于同時(shí)參與了多筆交易,你必須在再次購買前出售掉之前的股票。
示例 3:
輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這種情況下, 沒有交易完成, 所以最大利潤為 0。
2.1 常規(guī)思路
題目允許可以多次買賣,因此可以當(dāng)天買進(jìn)當(dāng)天賣出或者是當(dāng)天賣出當(dāng)天買進(jìn)。因此只要后一天的價(jià)格比前一天的價(jià)格高,就可以進(jìn)行買賣。即設(shè)置一個(gè)變量,初始化為0,當(dāng)后一天的價(jià)格比前一天的價(jià)格高時(shí),變量累加兩天價(jià)格的差值。
2.1 python常規(guī)代碼
class Solution:def maxProfit(self, prices: List[int]) -> int:if len(prices) == 0:return 0profit = 0for i in range(len(prices)-1):if prices[i+1] > prices[i]:profit += (prices[i+1] - prices[i])return profit2.2 動(dòng)態(tài)規(guī)劃
2.2 動(dòng)態(tài)規(guī)劃代碼
class Solution:def maxProfit(self, prices: List[int]) -> int:if len(prices) == 0:return 0n = len(prices)dp = [[0 for i in range(2)] for i in range(n)]dp[0][0] = 0dp[0][1] = -prices[0]for i in range(1,n):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[n-1][0]優(yōu)化代碼
可以對(duì)上面動(dòng)態(tài)規(guī)劃的空間復(fù)雜度進(jìn)行優(yōu)化。
class Solution:def maxProfit(self, prices: List[int]) -> int:if len(prices) == 0:return 0n = len(prices)dp_i_0 = 0dp_i_1 = -prices[0]for i in range(1,n):dp_i_0 = max(dp_i_0,dp_i_1+prices[i])dp_i_1 = max(dp_i_1,dp_i_0-prices[i])return dp_i_0總結(jié)
以上是生活随笔為你收集整理的leetcode:买卖股票的最佳时机2(python)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: h3c防火墙服务器ip修改,H3C防火墙
- 下一篇: 基于python实现开机自动连接校园网(