Leetcode. 2866.美丽塔II

最新推荐文章于 2024-07-26 01:21:03 发布

灬德布罗意的猫灬

最新推荐文章于 2024-07-26 01:21:03 发布

阅读量121

点赞数 2

分类专栏：单调栈文章标签： leetcode 算法

本文链接：https://blog.csdn.net/m0_60921016/article/details/134071343

版权

单调栈专栏收录该内容

2 篇文章 0 订阅

订阅专栏

要求O（N）复杂度内解决，考虑单调栈，这个题很像经典的美丽度的那个单调栈的模板题

对有每一个位置，考虑右边能扩展到哪来？不如直接从末尾来倒着看，发现从末尾需要维护一个单调增的单调栈，单调栈里面存下标就好了很经典

同理左边能扩展到哪里？也是直接从开头直接开始看就行了

这个题目唯一需要注意的一点就是我们一开始处理边界的时候可以放一个n（算右边能扩展的时候）同理算左边的时候你直接放一个-1 注意不要RE访问-1就好了

算是很板的一道题目

using ll = long long;
class Solution {
public:
    long long maximumSumOfHeights(vector<int>& maxHeights) {
        int n = maxHeights.size();
        vector<ll>left(n+10,0);
        vector<ll>right(n+10,0);
        stack<ll>stk;
        stk.push(n);
        for(int i=n-1;i>=0;i--){
            ll tem = maxHeights[i];
            while(stk.size()>1&&tem<=maxHeights[stk.top()])stk.pop();
            right[i] = right[stk.top()];
            right[i]+=(stk.top()-i)*tem;

            stk.push(i);
        }



        while(stk.size())stk.pop();
        stk.push(-1);
        for(int i=0;i<n;i++){
            ll tem = maxHeights[i];
            while(stk.size()>1&&tem<=maxHeights[stk.top()])stk.pop();
            
            if(stk.size()>1) left[i] = left[stk.top()];
            else left[i] =0; 
            left[i]+=(i-stk.top())*tem;

            stk.push(i);
        }

        ll ans = 0;
        for(int i=0;i<n;i++)
         ans = max(ans,left[i] + right[i]-maxHeights[i]);

        return ans;


    }
};