0
点赞
收藏
分享

微信扫一扫

LeetCode 2104. 子数组范围和***

看山远兮 2022-04-29 阅读 68
leetcode

基本思想:

和之前的题目一样,但是是升级版;

这个问题能转换一下:
对于所有可能序列的最大最小差,其实是所有序列的最大值和减去所有序列的最小值和;

因此可以转换为两次单调栈问题;

基本代码:

using ll=long long;
class Solution {
public:
    ll subArrayRanges(vector<int>& nums) {
        ll ret=0;
        int n=nums.size();
        vector<int>minx(n,0);
        vector<int>maxn(n,0);
        nums.push_back(INT_MIN);
        stack<int>minst;
        stack<int>maxst;
        for(int i=0;i<nums.size();i++){
            while(!minst.empty()&&nums[minst.top()]>=nums[i]){
                int index=minst.top();
                minst.pop();
                int right=i-index;
                int l=minst.empty()?-1:minst.top();
                int left=index-l;
                minx[index]=left*right;
            }
            minst.push(i);
        }
        nums.pop_back();
        nums.push_back(INT_MAX);
        for(int i=0;i<nums.size();i++){
            while(!maxst.empty()&&nums[maxst.top()]<=nums[i]){
                int index=maxst.top();
                maxst.pop();
                int right=i-index;
                int l=maxst.empty()?-1:maxst.top();
                int left=index-l;
                maxn[index]=left*right;
            }
            maxst.push(i);
        }
        for(int i=0;i<minx.size();i++){
            ret+=ll(nums[i])*maxn[i];
            ret-=ll(nums[i])*minx[i];
        }
        return ret;
    }
};
举报

相关推荐

0 条评论