leetcode-1732. 找到最高海拔

74 阅读1分钟

题目要求:

有一个自行车手打算进行一场公路骑行,这条路线总共由 n + 1 个不同海拔的点组成。自行车手从海拔为 0 的点 0 开始骑行。

给你一个长度为 n 的整数数组 gain ,其中 gain[i] 是点 i 和点 i + 1 的 净海拔高度差0 <= i < n)。请你返回 最高点的海拔 。

示例 1:
输入: gain = [-5,1,5,0,-7]
输出: 1
解释: 海拔高度依次为 [0,-5,-4,1,1,-6] 。最高海拔为 1 。

示例 2:
输入: gain = [-4,-3,-2,-1,4,3,2]
输出: 0
解释: 海拔高度依次为 [0,-4,-7,-9,-10,-6,-3,-1] 。最高海拔为 0 。

提示
n == gain.length
1 <= n <= 100`
-100 <= gain[i] <= 100

链接leetcode.cn/problems/fi…

解题思路

使用切片elevationList保存计算的海拔高度,海拔高度起始位为0,之后每个位次的高度均为海拔高度与海拔高度差之和。求出海拔高度的信息后,只需遍历求出最高海拔即可。时间复杂度是O(N+1)(海拔高度比gain多一个元素),空间复杂度为O(1)(只有maxHigh一个变量)

代码如下

func largestAltitude(gain []int) int {
	// 使用新的切片存高度
	// 遍历高度数据,返回最大值
	elevationList := make([]int, 0)
	elevationList = append(elevationList, 0)
	// 计算海拔高度
	for i := 0; i < len(gain); i++ {
		elevationList = append(elevationList, elevationList[i]+gain[i])
	}
	maxHigh := 0
	for i := 0; i < len(elevationList); i++ {
		if elevationList[i] > maxHigh {
			maxHigh = elevationList[i]
		}
	}
	return maxHigh
}

代码优化后

func largestAltitude(gain []int) int {
	// 前缀和
	ans := float64(0)
	total := 0
	for _, x := range gain {
		total += x
		ans = math.Max(float64(ans), float64(total))
	}
	return int(ans)
}

或者

func largestAltitude(gain []int) int {
	// 前缀和
	ans := 0
	total := 0
	for _, x := range gain {
		total += x
		if total > ans {
			ans = total
		}
	}
	return ans
}