ijd8.COM

A simple blog for an old Ma Nong.

Golang求出买卖股票的最佳时期

Permalink

给定一个数组,它的第 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。

提示:

  • 1 <= prices.length <= 3 * 10 ^ 4

  • 0 <= prices[i] <= 10 ^ 4

题解

本题其实就是找出数组中的递增子序列,然后将递增子序列的最后一个值减去第一个值的和的累加就是结果

所以我们给定一个数组prices,定义首尾两个指针

head=0 tail=1

算法如下

  • 如果 prices[j]>=prices[j-1] 那么说明当前的子序列是递增的,所以分一下两种情况处理
  • 如果j不是数组prices的最后一个值的位置,那么就不做处理,直接将j后移一位
  • 如果j是数组prices的最后一个值的位置,那么就用这个值减去位置i对应的值,并将这个结果累加到收益上面
  • 如果 prices[j]<prices[j-1],说明了递增序列已经结束了,那么做如下处理
  • 将差值 prices[j-1]-prices[i] 累加到收益上面,并且将i设置为值j,然后j继续向后遍历

代码如下:

Go: maxProfit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
func maxProfit(prices []int) int {
	sum := 0
	// 找出所有的递增(非减)序列
	for i, j := 0, 1; j < len(prices); j++ {
		if prices[j] >= prices[j-1] {
			if j == len(prices)-1 {
				sum += prices[j] - prices[i]
			}
			continue
		}
		sum += prices[j-1] - prices[i]
		i = j
	}
	return sum
}

Write a Comment

Submit Comment Login
Based on Golang + fastHTTP + sdb | go1.16.7 Processed in 0ms