Codeforces1037F Maximum Reduction 【单调栈】

题目分析：

没啥好说的，会单调栈就会做。

代码：

#include<bits/stdc++.h>
using namespace std;

const int maxn = 1005000;
const int mod = 1000000007;

int n,k;
int pre[maxn],suf[maxn],a[maxn];

stack <int> sta;
void getpre(){
    for(int i=n-1;i>=0;i--){
    while(!sta.empty()&&a[sta.top()]<=a[i])pre[sta.top()]=i+1,sta.pop();
    sta.push(i);
    }
    while(!sta.empty()) pre[sta.top()]=0,sta.pop();
}
void getsuf(){
    for(int i=0;i<n;i++){
    while(!sta.empty()&&a[sta.top()]<a[i])suf[sta.top()]=i-1,sta.pop();
    sta.push(i);
    }
    while(!sta.empty()) suf[sta.top()]=n-1,sta.pop();
}

void read(){
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    getpre(); getsuf();
}

int getsize(int len){
    int p = (len-1)/k;
    int ans = (1ll*len*p%mod-1ll*(1+p)*p/2%mod*k%mod+mod)%mod;
    return ans;
}

void work(){
    int ans = 0;k--;
    for(int i=0;i<n;i++){
    int tot = getsize(suf[i]-pre[i]+1);
    tot -= getsize(suf[i]-i); tot += mod; tot %= mod;
    tot -= getsize(i-pre[i]); tot += mod; tot %= mod;
    ans += 1ll*tot*a[i]%mod; ans %= mod;
    }
    printf("%d",ans);
}

int main(){
    read();
    work();
    return 0;
}