鉴于早上那题让我怀疑单调栈白学,特意来复习下单调栈

题意

[Agc005D/At2060] Minimum Sum - 单调栈

考虑按照每个元素对答案的贡献来统计,那么我们只需要找到每个元素左边右边第一个比它小的就可

这题给的又是排列,简直不能再良心

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1000005;

int s[N],top,a[N],l[N],r[N],n,ans;

signed main() {
    cin>>n;
    for(int i=1;i<=n;i++) {
        cin>>a[i];
    }
    for(int i=1;i<=n;i++) {
        while(top && a[i]<a[s[top]]) r[s[top]]=i, --top;
        s[++top]=i;
    }
    while(top) r[s[top]]=n+1, --top;
    for(int i=n;i>=1;--i) {
        while(top && a[i]<a[s[top]]) l[s[top]]=i, --top;
        s[++top]=i;
    }
    while(top) l[s[top]]=0, --top;
    for(int i=1;i<=n;i++) {
        ans += a[i] * (r[i]-i) * (i-l[i]);
    }
    cout<<ans;
}

相关文章:

  • 2022-12-23
  • 2021-08-20
  • 2021-11-26
  • 2022-01-10
  • 2021-05-27
  • 2021-09-02
猜你喜欢
  • 2021-08-02
  • 2021-12-17
  • 2022-02-21
  • 2022-12-23
相关资源
相似解决方案