Leetcode 买卖股票的最佳时机 II

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。

注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

示例 1:

1
2
3
4
输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4
随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3

示例 2:

1
2
3
4
5
输入: [1,2,3,4,5]
输出: 4
解释: 在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4
注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。
因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。

示例 3:

1
2
3
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

解决方案

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
func maxProfit(prices []int) int {
// 初始化变量
// tmpBuy 将要买入的价格
// tmpSell 将要卖出的价格
// buy 实际买入的价格
// sell 实际卖出的价格
// profit 利润
tmpBuy, tmpSell, buy, sell, profit := -1, -1, -1, -1, 0
for key, price := range prices {
// 初始化第一个元素为将要买入的价格
if key == 0 {
tmpBuy = price
continue
}

// 如果是第一次开始或者是刚卖出, 那么 buy == -1
if buy == -1 {
// 如果当前价格比想要买入的价格要低, 那么将当前价格设置成买入价格
if price <= tmpBuy {
tmpBuy = price
} else {
// 如果当前价格大于想要买入的价格, 那么将想要买入的价格直接买入,
// 并且将当前价格设置为想要出售的价格, 看看下面的价格是否能将利益最大化
// 如果不能的话, 那么就用这个价格作为出售价格
buy = tmpBuy
tmpSell = price
}
continue
}

// 如果已经买入了
if buy > -1 {
// 看一下价格是否大于想要售出的价格, 如果大于的话, 那么将当前价格设置为想要售出的价格
// 能够将利益最大化
if price >= tmpSell {
tmpSell = price
} else {
// 当前利益是最大化的, 直接售出, 并且将状态初始化
sell = tmpSell
profit = profit + (sell - buy)
tmpBuy = price
buy, sell, tmpSell = -1, -1, -1
}
}
}

// 如果买入了之后直到循环结束都没有卖出, 说明价格是递增的, 把最后一次的价格作为卖出价卖掉
if buy > -1 && sell == -1 {
profit += tmpSell - buy
}

return profit
}

Test

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
func TestMaxProfit(t *testing.T) {
cases := []struct {
in []int
want int
}{
{[]int{7, 1, 5, 3, 6, 4}, 7},
{[]int{1, 2, 3, 4, 5}, 4},
}
for _, c := range cases {
got := MaxProfit(c.in)
if got != c.want {
t.Errorf("maxProfit(%q) == %q, want %q", c.in, got, c.want)
}
}
}

提交结果:

image-20180805180624978