codeforces 1037F. Maximum Reduction(啟發式合併)
阿新 • • 發佈:2018-11-22
題目:http://codeforces.com/contest/1037/problem/F
思路:
找出最大的一個點(值相同時取左邊的點),計算以此點為最大值
能夠形成多少個合法的區間,然後處理該點左右兩個區間,一直遞迴下去。
假設以i為左端點,合法的右端點有i+(k-1),i+2*(k-1),i+3*(k-1),i+4*(k-1)
i+t*(k-1)不能超過區間的範圍。
#include<bits/stdc++.h> using namespace std; #define LL long long const LL mod=1e9+7; const int maxn=1e6+10; int a[maxn],n,k; int ll[maxn],rr[maxn],que[maxn]; LL ans; void init() { int top=0; for(int i=1;i<=n;i++) { while(top&&a[que[top]]<a[i]) top--; ll[i]=que[top]+1; que[++top]=i; } top=0;que[top]=n+1; for(int i=n;i>=1;i--) { while(top&&a[que[top]]<=a[i]) top--; rr[i]=que[top]-1; que[++top]=i; } } int main() { scanf("%d%d",&n,&k); k--; for(int i=1;i<=n;i++) scanf("%d",&a[i]); init(); for(int i=1;i<=n;i++) { int l=ll[i],r=rr[i]; if(2*i<=l+r) { for(int j=l;j<=i;j++) { int x=(i-j-1)/k; int y=(r-j)/k; x=max(0,x); if(y>x) ans+=1LL*(y-x)*a[i]; if(ans>=mod) ans%=mod; } } else { for(int j=i;j<=r;j++) { int x=(j-i-1)/k; int y=(j-l)/k; x=max(0,x); if(y>x) ans+=1LL*(y-x)*a[i]; if(ans>=mod) ans%=mod; } } } printf("%lld\n",ans%mod); return 0; }