Submit 309_BestTimetoBuyandSellStockwithCooldown

Binlogo-Archive · Jul 10, 2020 · 0344720 · 0344720
1 parent 18e5f2a
commit 0344720
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/...layground/Pages/309_BestTimetoBuyandSellStockwithCooldown.xcplaygroundpage/Contents.swift b/...layground/Pages/309_BestTimetoBuyandSellStockwithCooldown.xcplaygroundpage/Contents.swift
@@ -0,0 +1,27 @@
+//: [Previous](@previous)
+
+import Foundation
+
+class Solution {
+    func maxProfit(_ prices: [Int]) -> Int {
+        guard !prices.isEmpty else { return 0 }
+        var rest = 0
+        var sold = 0
+        var hold = Int.min
+        for current in 0..<prices.count {
+            let currentPrice = prices[current]
+            let preSold = sold
+            let preHold = hold
+            hold = max(preHold, rest - currentPrice)
+            sold = preHold + currentPrice
+            rest = max(rest, preSold)
+        }
+        return max(rest, sold)
+    }
+}
+
+// Tests
+let s = Solution()
+s.maxProfit([1,2,3,0,2]) == 3
+
+//: [Next](@next)