2866. 美丽塔 II(2072)

给你一个长度为 n 下标从 0 开始的整数数组 maxHeights

你的任务是在坐标轴上建 n 座塔。第 i 座塔的下标为 i ,高度为 heights[i]

如果以下条件满足,我们称这些塔是 美丽 的:

  1. 1 <= heights[i] <= maxHeights[i]
  2. heights 是一个 山脉 数组。

如果存在下标 i 满足以下条件,那么我们称数组 heights 是一个 山脉 数组:

  • 对于所有 0 < j <= i ,都有 heights[j - 1] <= heights[j]
  • 对于所有 i <= k < n - 1 ,都有 heights[k + 1] <= heights[k]

请你返回满足 美丽塔 要求的方案中,高度和的最大值

示例 1:

1
2
3
4
5
6
输入:maxHeights = [5,3,4,1,1]
输出:13
解释:和最大的美丽塔方案为 heights = [5,3,3,1,1] ,这是一个美丽塔方案,因为:
- 1 <= heights[i] <= maxHeights[i]
- heights 是个山脉数组,峰值在 i = 0 处。
13 是所有美丽塔方案中的最大高度和。

示例 2:

1
2
3
4
5
6
输入:maxHeights = [6,5,3,9,2,7]
输出:22
解释: 和最大的美丽塔方案为 heights = [3,3,3,9,2,2] ,这是一个美丽塔方案,因为:
- 1 <= heights[i] <= maxHeights[i]
- heights 是个山脉数组,峰值在 i = 3 处。
22 是所有美丽塔方案中的最大高度和。

示例 3:

1
2
3
4
5
6
7
输入:maxHeights = [3,2,5,5,2,3]
输出:18
解释:和最大的美丽塔方案为 heights = [2,2,5,5,2,2] ,这是一个美丽塔方案,因为:
- 1 <= heights[i] <= maxHeights[i]
- heights 是个山脉数组,最大值在 i = 2 处。
注意,在这个方案中,i = 3 也是一个峰值。
18 是所有美丽塔方案中的最大高度和。

提示:

  • 1 <= n == maxHeights <= 10^5
  • 1 <= maxHeights[i] <= 10^9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class Solution {
public long maximumSumOfHeights(List<Integer> maxHeights) {
int n = maxHeights.size();
long l[] = new long[n+1], r[] = new long[n+1];
Deque<int[]> dq = new LinkedList<>();
for (int i = 0; i < n; i++) {
int cur = maxHeights.get(i);
while (!dq.isEmpty() && dq.peek()[0] > cur) {
dq.pop();
}
int rightMostIdx = dq.isEmpty() ? -1 : dq.peek()[1];
l[i+1] = (long)cur*(i-rightMostIdx) + l[rightMostIdx+1];
dq.push(new int[]{cur, i});
}
dq.clear();
for (int i = n-1; i >= 0; i--) {
int cur = maxHeights.get(i);
while (!dq.isEmpty() && dq.peek()[0] > cur) {
dq.pop();
}
int leftMostIdx = dq.isEmpty() ? n : dq.peek()[1];
r[i] = (long)cur*(leftMostIdx-i) + r[leftMostIdx];
dq.push(new int[]{cur, i});
}
long res = 0;
for (int i = 0; i < n; i++) {
res = Math.max(res, l[i+1] + r[i] - maxHeights.get(i));
}
return res;
}
}