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

链接: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 做法比较好想到
  将上面的式子再进行一步化简可以得到的任意一个区间的最大值总和 - 任意一个区间最小值的总和
  对于上式的两部分我们可以分别去求,用单调栈即可
  例如在求最大值的和累加时,去维护一个单调递减的栈,具体看代码就懂了
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#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;
}

 

作者:静默虚空
欢迎任何形式的转载,但请务必注明出处。
限于本人水平,如果文章和代码有表述不当之处,还请不吝赐教。

posted @   楼主好菜啊  阅读(3998)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 25岁的心里话
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 零经验选手,Compose 一天开发一款小游戏!
· 因为Apifox不支持离线,我果断选择了Apipost!
· 通过 API 将Deepseek响应流式内容输出到前端
历史上的今天:
2017-11-02 dijkstra - 堆优化
2017-11-02 重载运算符-operator
2017-11-02 最短路 - dijkstra
点击右上角即可分享
微信分享提示