結果

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

ソースコード

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];
    }
    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 = floor(sum[i+1]*1.0/(i+1));
        u = floor((sum[n]-sum[i+1])*1.0/(n-i-1));
        // std::cout << sum[i+1]*1.0<<" "<< (sum[n]-sum[i+1])<< std::endl;

        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);
        
        // std::cout << d<<" "<<dv<<" "<<u<<" "<<uv << std::endl;
        // -1 10 12 14
        result = min(dv+uv,result);
    }
    // 70 6
    // 11.6666
    std::cout << result << std::endl;
}
0