2024.1.24
题目来源
我的题解
方法一 暴力枚举
将每个位置都作为山峰来进行遍历,计算每个山峰下的最大山脉数组和
时间复杂度:O( n 2 n^2 n2)
空间复杂度:O(1)
public long maximumSumOfHeights(List<Integer> maxHeights) {
int n=maxHeights.size();
long res=0;
for(int i=0;i<n;i++){
res=Math.max(res,getSum(maxHeights,i));
}
return res;
}
public long getSum(List<Integer> maxHeights,int index){
long res=maxHeights.get(index);
int t=maxHeights.get(index);
for(int i=index-1;i>=0;i--){
int cur=maxHeights.get(i);
if(cur<=t){
res+=cur;
t=cur;
}else{
res+=t;
}
}
t=maxHeights.get(index);
for(int i=index+1;i<maxHeights.size();i++){
int cur=maxHeights.get(i);
if(cur<=t){
res+=cur;
t=cur;
}else{
res+=t;
}
}
return res;
}
方法二 单调栈+前、后缀和
首先需要知道山状数组是会从山顶将数组分为两个部分,数组左侧构成非递减,数组右侧构成非递增。想要使得数组元素尽可能大,需要使得heights[i]取值为maxHeights[i],此时假设区间[0,i]构成的非递减元素和最大值为prefix[i],区间[i,n-1]构成的非递增数组元素和的最大值为suffix[i],则这时的山状数组的元素之和为:prefix[i]+suffic[i]-maxHeights[i]。减去maxHeights[i]是因为maxHeights[i]在左侧和右侧都被计算进去了,也就是计算了两次,所以需要减去一次。接下来分别讨论左侧和右侧的前缀和以及后缀和的计算:
- 左侧的非递减(求前缀和):将maxHeights依次入栈,对于i位置元素来说,不断从栈顶弹出元素,直到栈中元素是一个非递减情况(也就是栈顶元素小于maxHeights[i])。假设栈顶元素为j位置元素,则对于i位置的最大前缀和:prefix[i]=prefix[j]+(i-j)*maxHeights[i]
- 右侧的非递增(求后缀和):将maxHeights依次入栈,对于i位置元素来说,不断从栈顶弹出元素,直到栈中元素是一个非递减情况(也就是栈顶元素小于maxHeights[i])。假设栈顶元素为j位置元素,则对于i位置的最大前缀和:suffix[i]=suffix[j]+(j-i)*maxHeights[i]
所以最终的山状数组的最大值为:max(prefix[i]+suffic[i]-maxHeights[i])
时间复杂度:O(n)
空间复杂度:O(n)
有任何问题,欢迎评论区交流,欢迎评论区提供其它解题思路(代码),也可以点个赞支持一下作者哈😄~