結果

問題 No.738 平らな農地
コンテスト
ユーザー vjudge1
提出日時 2026-02-23 15:31:17
言語 C++14
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++14 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 1,567 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,174 ms
コンパイル使用メモリ 193,288 KB
実行使用メモリ 15,368 KB
最終ジャッジ日時 2026-02-23 15:31:26
合計ジャッジ時間 7,261 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 87
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1000010;
const int INF=0x3f3f3f3f3f3f3f3f;
int n,k;
int ans=INF;
int a[N],b[N];
struct BIT{
    int t[N];
    int lowbit(int x){return x&-x;}
    void update(int x,int y){
        for(int i=x;i<=n;i+=lowbit(i))t[i]+=y;
        return;
    }
    int ask(int x){
        int res=0;
        for(int i=x;i>=1;i-=lowbit(i))res+=t[i];
        return res;
    }
    int query(int l,int r){
        return ask(r)-ask(l-1);
    }
}T,S;
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
    // freopen("farmland.in","r",stdin);
    // freopen("farmland.out","w",stdout);
    cin>>n>>k;
    vector<int>v;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[i]=a[i];
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=1;i<=n;i++)a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
    for(int i=1;i<=n;i++){
        T.update(a[i],1);
        S.update(a[i],b[i]);
        if(i<k)continue;
        if(i>k){
            T.update(a[i-k],-1);
            S.update(a[i-k],-b[i-k]);
        }
        int num=0;
        int l=1,r=n,mid;
        while(l<=r){
            mid=(l+r)>>1;
            if(T.ask(mid)>=(k+1)/2){
                num=mid;
                r=mid-1;
            }
            else l=mid+1;
        }
        int L=v[num-1]*T.ask(num)-S.ask(num);
        int R=(S.ask(n)-S.ask(num))-v[num-1]*(T.ask(n)-T.ask(num));
        ans=min(ans,L+R);
    }
    cout<<ans<<"\n";
    return 0;
}
/*

*/
0