求任意一个区间中的最大值,最小值 - 单调栈
链接:https://ac.nowcoder.com/acm/contest/223/C
来源:牛客网
题目描述
给出长度为n的序列a,其中第i个元素为,定义区间(l,r)的价值为
请你计算出
输入描述:
第一行输入数据组数T
对于每组数据,第一行为一个整数n,表示序列长度
接下来一行有n个数,表示序列内的元素
输出描述:
对于每组数据,输出一个整数表示答案
示例1
备注:
不保证数据随机生成!
题意 : 题目经过几步简单的化简便可以知道我们要求的是任意一个区间中的最大值与最小值的差去累加
思路分析:暴力的 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; } |
作者:静默虚空
欢迎任何形式的转载,但请务必注明出处。
限于本人水平,如果文章和代码有表述不当之处,还请不吝赐教。
东北日出西边雨 道是无情却有情
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步