1. 程式人生 > >121、買賣股票的最佳時機

121、買賣股票的最佳時機

###### 121、買賣股票的最佳時機 給定一個數組,它的第`i` 個元素是一支給定股票第 `i` 天的價格。 如果你最多隻允許完成一筆交易(即買入和賣出一支股票一次),設計一個演算法來計算你所能獲取的最大利潤。 注意:你不能在買入股票前賣出股票。 ``` 示例 1: 輸入: [7,1,5,3,6,4] 輸出: 5 解釋: 在第 2 天(股票價格 = 1)的時候買入,在第 5 天(股票價格 = 6)的時候賣出,最大利潤 = 6-1 = 5 。 注意利潤不能是 7-1 = 6, 因為賣出價格需要大於買入價格;同時,你不能在買入前賣出股票。 ``` ``` 示例 2: 輸入: [7,6,4,3,1] 輸出: 0 解釋: 在這種情況下, 沒有交易完成, 所以最大利潤為 0。 ``` > 第一種方法:暴力法 ```python """ 直接遍歷迴圈求解 """ class Solution: def maxProfit(self, prices): n = len(prices) if n <= 1: return 0 max_profit = 0 for i in range(n): for j in range(i, n): if prices[i] < prices[j]: profit = prices[j] - prices[i] max_profit = max(profit, max_profit) return max_profit ``` > 時間複雜度`O(n^2)`,空間複雜度`O(1)` 從上面執行時間超時,放到pycharm中執行結果正確,因此想辦法將時間降低。 > 第二種辦法:動態規劃 > > > 動態規劃步驟, > > > > > > 1、明確`dp[i]`表示的是什麼關係 > > > > > > 2、找到`dp[i]`與`dp[i - 1]`的關係得出狀態轉移方程 > > > > > > 3、確定初始條件,`dp[0]` 在這裡,1、`dp[i]`表示的是第`i`天的最大利潤。2、`dp[i]`與`dp[i - 1]`的關係得出狀態轉移方程是`dp[i] = max(dp[i-1], price - min_price)`(也就是前一天的利潤與今天的利潤最大的一個)。3、`dp[0]`表達的是`0`。 程式碼實現如下: ```python class Solution: def maxProfit(self, prices): n = len(prices) if n == 0 or n <= 1:return 0 dp = [0]*n #初始化dp[0] = 0了 min_price = prices[0] for i in range(1, n): min_price = min(min_price, pries[i] dp[i] = max(dp[i - 1], prices[i] - min_price) return dp[-1] ``` > 時間複雜度`O(n)`,空間複雜度O(n)。 > 第三種辦法:一次遍歷解決(對動態規劃的優化) ```python class Solution: def maxProfit(self, prices): if not prices or len(prices) < 2:return 0 min_price = float("inf") max_profit = 0 for price in prices: min_price = min(min_price, price) max_profit = max(max_profit, price - min_price) return max_profit ``` > 時間複雜度`O(n)`,空間複雜