单调栈:区区区间间间(寻找任意区间的最大值和最小值)

题目描述:

链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
 

示例1

输入

3
3
4 2 3
5
1 8 4 3 9
20
2 8 15 1 10 5 19 19 3 5 6 6 2 8 2 12 16 3 8 17 

输出

5

57

2712

说明

对于一组测试数据的解释:
区间[1, 2]的贡献为:4 - 2 = 2
区间[1, 3]的贡献为:4 - 2 = 2
区间[2, 3]的贡献为:3 - 2 = 1
2 + 1 + 2 = 5.

备注:

 

T⩽20,n⩽10^5,0⩽ai⩽10^5 

不保证数据随机生成!

思路:(来自大佬)

先对式子进行简化,可以变成求区间所有最大值之和减去区间所有最小值之和。那么关键就在于求出所有区间的最大值以及最小值。但是如果光暴力的话(枚举区间长度+区间平移),是会超时的,那么可以选择换一种策略,选择对数组的元素进行枚举,那么建立两个数组l、r,则可以表示出以它作为最大值的区间的下界l【i】、上界r【i】,那么对于每一个枚举出来的元素(假设为mas【i】,就可以求出以它为最大值的区间的最大值的总和(最大值,即该元素的值,乘以所有满足条件的区间的个数),应该分为两部分,一种情况:i是区间端点,比如2 3 3 4 ,4作为区间端点。那么它所对应的值就应该是mas【i】*(r【i】-l【i】),即最大值*区间长度。还有另一种情况,该元素在区间内部,所有满足条件的区间的个数就应该是其左边能取的元素的个数(作为下界)*其右边能取的元素的个数(作为上界),即:mas【i】*(r【i】-i)*(i-l【i】);对所有元素遍历,将其两个部分相加,就得到所有最大值的总和了。那么对于最小值,只要把所有元素取反,再进行同样操作就可以了(因为元素取反后,上下界也会变化并改变大小关系,那么两个负的相乘,结果还是正的);

还有几个细节要注意,因为mas【i】是有可能作为区间端点存在的,所以 在求l【i】、r【i】时,枚举的j应该定义为i,而不是i+1或i-1。然后是取等问题,象3 1 4 4 4这个数组,三个4是重复的,都可以作为同一个区间的最大值(最小值),显然,求上界、下界两种情况时,如果我们不考虑等号的话,情况是会有遗漏的,比如遗漏3 1 4 4 这个区间(如果我们选第一个4作为最大值)。但是等号也不能全加,否则,以第一个4作为最大值时,会包括3 1 4 4 4 这个区间,而以第二个4作为最大值时,也会把这个3 1 4 4 4区间再次包含,那么情况就重了。那么应该只加一个等号,那么加哪个呢?其实这个等号就是代表了一种方向,即在不加等号的那种枚举里(求上界/下界),求的是满足这个方向内满足严格小于该元素的区间的端点,所以其实等号加在那个都没事,只要加了就行。还是上面那个例子,如果求下界时可以取等,但求上界时要求严格小于,则:当我们取第一个4作为区间最大值时,只能枚举到3 1 4这个区间,而以第二个4作为最大值时,就可以包括3 1 4 4,而不会有3 1 4 这个区间,也不会有3 1 4 4 4这个区间,这样就确保了对于每个元素,其区间的唯一性。最后一个,是在求区间上界时的枚举顺序,应该是逆序。

r【i】返回的是对于第i个元素,下标最大的不大于它的元素的下标,刚进入循环时,j=n,退出,r【n】=n;然后j=n-1,如果mas【n-1】满足条件的话,j=r【n-1+1】=n;如果不满足,就这样依次向下取。逆序枚举正是为了让一开始的r数组有具体的数字,如果正着枚举的话,就会陷入死循环,因为r的任意元素都没有值。

代码: 

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll t,n,mas[100010],sum=0;
ll l[100010],r[100010];//l:区间下界,l[i]表示下标;r:区间上界 
ll f(){
    //ll l[100010],r[100010];//l:区间下界,l[i]表示下标;r:区间上界 
    for(int i=1;i<=n;++i){
        int j=i;//i有可能作为区间端点,即l[i]==i; 
        while((j>1)&&(mas[j-1]<mas[i]))
            j=l[j-1];//迭代向下; 
        l[i]=j;
    
    }
    for(int i=n;i>=1;--i){
        int j=i;
        while((j<n)&&(mas[j+1]<=mas[i]))
            j=r[j+1];
            r[i]=j;
        
        
    }
    ll ans=0;
    for(int i=1;i<=n;++i){
        ans+=mas[i]*(r[i]-l[i]);
        ans+=mas[i]*(r[i]-i)*(i-l[i]);
    }
    return ans;
}
int main()

{
    cin>>t;
    while(t--){
        //sum=0;
        cin>>n;
        for(int i=1;i<=n;++i) cin>>mas[i];
        sum=f();
        for(int i=1;i<=n;++i) mas[i]=-mas[i];
        sum+=f();
        cout<<sum<<endl;
    }
    return 0;
}

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值