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

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

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

示例 1:

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

示例 3:

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

解决方案

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

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