結果

問題 No.837 Noelちゃんと星々2
ユーザー ngtkana
提出日時 2020-03-13 00:43:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,347 bytes
コンパイル時間 2,136 ms
コンパイル使用メモリ 199,696 KB
最終ジャッジ日時 2025-01-09 06:34:09
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 3 WA * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define endl enjoy_codeforces
using lint=long long;
void cmn(lint&x,lint y){if(x>y)x=y;}
lint round_div(lint x, lint y){return x/y+(y/2<x%y);}
int main(){
    std::cin.tie(nullptr);std::ios_base::sync_with_stdio(false);
    int n;std::cin>>n;
    std::vector<lint>a(n);
    for(lint&x:a)std::cin>>x;
    std::sort(a.begin(),a.end());
    if(a.front()==a.back()){
        std::cout<<n/2<<'\n';
        return 0;
    }
    lint N=std::accumulate(a.begin(),a.end(),0ll);
    lint L=0,C=a.at(0),R=a.at(0);
    lint ans=N;
    for(int c=1,l=0,r=1;c<n;C+=a.at(c++)){
        lint lave=round_div(C,c);
        lint rave=round_div(N-C,n-c);
        for(;l<c&&a.at(l)<lave;L+=a.at(l++));
        for(;r<n&&a.at(r)<rave;R+=a.at(r++));
        cmn(ans,
            lave*l-L
            +(C-L)-lave*(c-l)
            +rave*(r-c)-(R-C)
            +(N-R)-rave*(n-r));
    }
    std::cout<<ans<<'\n';
}

/*
 * ひとつにまとめるのであれば、
 * 平均の切り上げ・切り下げのうち近いほうです。
 *
 * 二つですから、ソートして境界を全探索です。
 * 左側と右側それぞれの合計を管理です。
 * すると、目標高さがわかります。
 * 目標高さより小さい部分と、高い部分の境界も管理です。
 *     L: [0, l[
 *     C: [0, c[
 *     R: [0, r[
 */
0