Codeforces 1037 F. Maximum Reduction —— 单调栈

This way

题意：

给你一个长度为n的数组和一个值k，ans=z(a,k)：

问你ans是多少

题解：

其实蛮简单的，稍微模几个样例就知道，其实就是第一次是取长度为k的最大值，第二次是取长度为k+k-1的最大值，第三次是取长度为k+k-1+k-1的最大值…

那么很明显这道题目就是枚举每个数，在他是最大的区间里面求有多少种情况。
对于在区间为$l-r$之内的请况用等差数列求和解决，那么如何又在$l-r$之内又要覆盖点i呢，这时候用正向思维去加我感觉是很复杂的，需要去减，也就是$Cal(l,r)-Cal(l,i-1)-Cal(mid+1,r)$
然后求所在最大的区间就用单调栈求

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e6+5;
const ll mod=1e9+7;
int n,k;
ll cal(int l,int r){
    int len=r-l+1;
    if(len<k)return 0;
    int num=(r-l+1-k+1)/(k-1);
    return 1ll*(num+1)*(len-k+1+len-k+1-num*(k-1))/2%mod;
}
ll a[N];
struct node{
    ll x;
    int id;
}st[N];
int top,l[N],r[N];
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++){
        while(top&&st[top].x<a[i])
            top--;
        l[i]=st[top].id+1;
        st[++top]={a[i],i};
    }
    top=0;
    for(int i=n;i;i--){
        while(top&&st[top].x<=a[i])
            top--;
        r[i]=top?st[top].id-1:n;
        st[++top]={a[i],i};
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
        ans=(ans+(cal(l[i],r[i])-cal(l[i],i-1)-cal(i+1,r[i])+2*mod)%mod*a[i])%mod;
    printf("%lld\n",ans);
    return 0;
}