1. 程式人生 > >codeforces 1037F. Maximum Reduction(啟發式合併)

codeforces 1037F. Maximum Reduction(啟發式合併)

題目: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;
}