1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
| // @Title: 最佳买卖股票时机含冷冻期 (Best Time to Buy and Sell Stock with Cooldown)
// @Author: 15816537946@163.com
// @Date: 2019-09-24 23:36:17
// @Runtime: 0 ms
// @Memory: 2.3 MB
/*
* @lc app=leetcode.cn id=309 lang=golang
*
* [309] 最佳买卖股票时机含冷冻期
*
* https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/description/
*
* algorithms
* Medium (49.76%)
* Likes: 137
* Dislikes: 0
* Total Accepted: 7.6K
* Total Submissions: 15.3K
* Testcase Example: '[1,2,3,0,2]'
*
* 给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
*
* 设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
*
*
* 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
* 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
*
*
* 示例:
*
* 输入: [1,2,3,0,2]
* 输出: 3
* 解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
*
*/
func maxProfit(prices []int) int {
n := len(prices)
if n == 0 {
return 0
}
buy := make([]int, n+1)
buy[1] = 0 - prices[0]
sel := make([]int, n+1)
for i := 2; i <= n; i++ {
buy[i] = max(buy[i-1], sel[i-2]-prices[i-1])
sel[i] = max(sel[i-1], buy[i-1]+prices[i-1])
}
return sel[n]
}
func max(a, b int) int {
if a > b {
return a
}
return b
}
|