首页 > 其他 > 详细

求任意一个区间中的最大值,最小值 - 单调栈

时间:2018-11-02 13:49:14      阅读:276      评论:0      收藏:0      [点我收藏+]

链接:https://ac.nowcoder.com/acm/contest/223/C
来源:牛客网

题目描述

给出长度为n的序列a,其中第i个元素为技术分享图片,定义区间(l,r)的价值为
技术分享图片
请你计算出技术分享图片

输入描述:

第一行输入数据组数T
对于每组数据,第一行为一个整数n,表示序列长度
接下来一行有n个数,表示序列内的元素

输出描述:

对于每组数据,输出一个整数表示答案
示例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.

备注:

技术分享图片
不保证数据随机生成!
 
题意 : 题目经过几步简单的化简便可以知道我们要求的是任意一个区间中的最大值与最小值的差去累加
思路分析:暴力的 n^2 做法比较好想到
  将上面的式子再进行一步化简可以得到的任意一个区间的最大值总和 - 任意一个区间最小值的总和
  对于上式的两部分我们可以分别去求,用单调栈即可
  例如在求最大值的和累加时,去维护一个单调递减的栈,具体看代码就懂了
#define ll long long
const ll maxn = 1e5+5;

ll n;
ll a[maxn], sta[maxn];

void solve() {
    ll top = 0;
    ll sum = 0, ans = 0;
    for(ll i = 1; i <= n; i++){ // 下降
        while(top > 0 && a[sta[top]] < a[i]) {
            ll num = a[sta[top]];
            sum -= num*(sta[top]-sta[top-1]);
            top--;
        } 
        sta[++top] = i;
        sum += a[i]*(sta[top]-sta[top-1]);
        ans += sum;
    }
    
    top = 0, sum = 0;
    for(ll i = 1; i <= n; i++){ // 上升
        while(top > 0 && a[sta[top]] > a[i]){
            ll num = a[sta[top]];
            sum -= num*(sta[top]-sta[top-1]);
            top--;
        }
        sta[++top] = i;
        sum += a[i]*(sta[top]-sta[top-1]);
        ans -= sum;
    }
    cout << ans << ‘\n‘;
}

int main() {
    ll t;
    
    cin >> t;
    while(t--){
        scanf("%lld", &n);
        for(ll i = 1; i <= n; i++){
            scanf("%d", &a[i]);
        }        
        solve();
    }
    return 0;
}

 

求任意一个区间中的最大值,最小值 - 单调栈

原文:https://www.cnblogs.com/ccut-ry/p/9895593.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!