golang 子数组最大乘积

leetcode题目152
输入 [2,3,-2,4] 返回 6
输入 [-2,0,-1] 返回 0

题目分析: 动态规划中等难度的题目，最大乘积具有后无效性
由于乘法具有负负得正的特性，所以需要记录当前最大与当前最小值，这样可以确保遇到负数的时候及时翻转结果。
记数组为a，最大乘积为imax，上一个最大乘积为preMax, 上一个最小乘积为preMin， 状态转移方程

$$imax[i]=max(preMax\ast a[i],preMin\ast a[i],a[i])$$

对应的代码是

package main

import "fmt"

func max(x, y int) int {
	if x > y {
		return x
	} else {
		return y
	}
}

func main() {
	var a = []int{2, -3, -2, 4}
	//var a = []int{-2,0,-1}

	// imax=max(a[i], preMax*a[i], preMin*a[i])

	preMax, preMin, imax := a[0], a[0], a[0]
	for i := 1; i < len(a); i++ {
		if a[i] < 0 {
			preMax, preMin = preMin, preMax
		}
		preMax = max(a[i]*preMax, a[i])
		preMin = max(a[i]*preMin, a[i])

		imax = max(preMax, preMin)

	}
	fmt.Println(imax)
}