题目要求:
有一个自行车手打算进行一场公路骑行,这条路线总共由 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
解题思路
使用切片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
}