LeetCode刷题：188. 买卖股票的最佳时机 IV（golang版）

• 题目描述

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

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1：
输入：k = 2, prices = [2,4,1]
输出：2
解释：在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。

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

提示：
0 <= k <= 100
0 <= prices.length <= 1000
0 <= prices[i] <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

• 思路看代码注释
  这道题是(123.买卖股票的最佳时机 III) 的升级版，区别是123题是固定交易2次，而这道题是的交易次数是变量K，搞懂这个题第123题自然也就解决了
• 代码

package main

import "fmt"

func maxProfit(k int, prices []int) int {
	if len(prices) == 0 {
		return 0
	}
	//go不能定义变长数组，所以这里直接定义k的最大值100+1空间大小的数组，后面可以再优化
	var dp = make([][101][2]int,len(prices))
	//初始化dp数组，第0天的k次卖出都为0，第0天的k次买入均为-prices[0]
	for i:=0; i<=k; i++ {
		dp[0][i][0] = 0
		dp[0][i][1] = -prices[0]
	}
	//还是动态规划思想，核心有两点，1是每个步骤的状态定义，2是状态间的转换方程，首先要理清这两点
	//复杂的是有了k次交易的限制，所以之前二维的dp不能再满足所有信息的记录，但再加一维即可记录所有信息。
	//每天的状态有两个，持有股票或不持有股票，这里，0：代表不持有股票，可以理解为卖出，1：持有股票，可以理解为买入
	//原先用dp[i][0]代表第i天不持有股票，dp[i][1]代表第i天持有股票，但这里需要考虑交易次数，所以，dp[i][ki][0]表示第i天第ki次交易不持有股票，dp[i][ki][1]表示第i天第ki次交易持有股票
	//dp[i][ki][0]的也是两种情况，1：第i-1天卖出，第i天不操作,这里需要考虑卖出次数ki，记上一次卖出时的ki为ki-1。2：第i-1天买入，然后第i天卖出，这里也要考虑卖出次数ki,但需要注意的是，买入时不用考虑ki，只有在卖出时才算一次完整的交易。
	//所以，dp[i][ki][0]=max(dp[i-1][ki-1][0],dp[i-1][ki][1]+prices[i]),但这里还有一种情况，这里计算出的dp[i][ki][0]并不总是历史最大值，所以需要与dp[i-1][ki][0]比较，将两者较大值重新赋值给dp[i-1][ki][0]即可
	//dp[i][ki][1]类似，也是两种情况，1:第i-1天买入，第i天不操作,买入不需要考虑交易此时ki，2：第i-1天卖出，然后第i天买入,卖出需要考虑交易次数ki，第i-1天卖出，交易次数应为ki-1。所以，dp[i][ki][1] = max(dp[i-1][ki][1],dp[i-1][ki-1][0]-prices[i])
	for i:=1; i<len(prices); i++ {
		for ki:=1; ki<=k; ki++ {
			dp[i][ki][0] = max(dp[i-1][ki][0],max(dp[i-1][ki-1][0],dp[i-1][ki][1]+prices[i]))
			dp[i][ki][1] = max(dp[i-1][ki][1],dp[i-1][ki-1][0]-prices[i])
		}
	}
	return dp[len(prices)-1][k][0]
}

func max(a int, b int) int {
	if a > b {
		return a
	}
	return b
}


func main() {
	//nums  := []int{3,3,5,0,0,3,1,4}
	nums  := []int{3,2,6,5,0,3}
	res := maxProfit(2,nums)
	fmt.Println(res)
}