結果

問題 No.837 Noelちゃんと星々2
ユーザー erbowl
提出日時 2019-08-18 08:30:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,048 bytes
コンパイル時間 1,840 ms
コンパイル使用メモリ 172,580 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-01 13:34:22
合計ジャッジ時間 4,248 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 4 WA * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i, n) for(int i = 0; i < (int)(n); i++)
typedef long long ll;
#include <bits/stdc++.h>
using namespace std;
int main() {
    ll n;
    std::cin >> n;
    vector<ll> y(n);
    for (int i = 0; i < n; i++) {
        std::cin >> y[i];
        y[i]+=1e9;
    }
    sort(y.begin(),y.end());
    
    if(y[0] == y[n-1]){
        std::cout << 1 << std::endl;
        return 0;
    }

    vector<ll> sum(n+1);
    for (int i = 0; i < n; i++) {
        sum[i+1] = sum[i]+y[i]; 
    }
    
    ll result = 1e15;
    
    for (int i = 0; i < n-1; i++) {
        ll d,u,ijou;
        d = round(sum[i+1]*1.0/(i+1));
        u = round((sum[n]-sum[i+1])*1.0/(n-i-1));

        ijou = distance(y.begin(),lower_bound(y.begin(),y.end(),d));
        ll dv = d*ijou-sum[ijou]+(sum[i+1]-sum[ijou])-d*(i+1-ijou);

        ijou = distance(y.begin(),lower_bound(y.begin(),y.end(),u));
        ll uv = u*(ijou-i-1)-(sum[ijou]-sum[i+1])+(sum[n]-sum[ijou])-u*(n-ijou);
        
        result = min(dv+uv,result);
    }
    
    std::cout << result << std::endl;
}
0