题意:
给你一个长度为n的数组和一个值k,ans=z(a,k):
问你ans是多少
题解:
其实蛮简单的,稍微模几个样例就知道,其实就是第一次是取长度为k的最大值,第二次是取长度为k+k-1的最大值,第三次是取长度为k+k-1+k-1的最大值…
那么很明显这道题目就是枚举每个数,在他是最大的区间里面求有多少种情况。
对于在区间为
l
−
r
l-r
l−r之内的请况用等差数列求和解决,那么如何又在
l
−
r
l -r
l−r之内又要覆盖点i呢,这时候用正向思维去加我感觉是很复杂的,需要去减,也就是
C
a
l
(
l
,
r
)
−
C
a
l
(
l
,
i
−
1
)
−
C
a
l
(
m
i
d
+
1
,
r
)
Cal(l,r)-Cal(l,i-1)-Cal(mid+1,r)
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;
}