結果

問題 No.837 Noelちゃんと星々2
ユーザー kimiyuki
提出日時 2019-06-14 21:59:39
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 25 ms / 2,000 ms
コード長 1,167 bytes
コンパイル時間 1,889 ms
コンパイル使用メモリ 199,572 KB
最終ジャッジ日時 2025-01-07 04:31:14
ジャッジサーバーID
(参考情報)
judge5 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:32:17: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   32 |     int n; scanf("%d", &n);
      |            ~~~~~^~~~~~~~~~
main.cpp:35:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   35 |         scanf("%lld", &a[i]);
      |         ~~~~~^~~~~~~~~~~~~~~

ソースコード

diff #

#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i))
#define REP3(i, m, n) for (int i = (m); (i) < (int)(n); ++ (i))
#define ALL(x) begin(x), end(x)
using ll = long long;
using namespace std;
template <class T, class U> inline void chmin(T & a, U const & b) { a = min<T>(a, b); }

ll solve(int n, vector<ll> & a) {
    sort(ALL(a));
    if (a.front() == a.back()) {
        return 1;
    }
    vector<ll> acc(n + 1);
    partial_sum(ALL(a), acc.begin() + 1);

    ll ans = LLONG_MAX;
    REP3 (m, 1, n) {
        ll sum = 0;
        int i = upper_bound(a.begin(), a.begin() + m, a[m / 2]) - a.begin();
        sum += i * a[m / 2] - (acc[i] - acc[0]);
        sum += (acc[m] - acc[i]) - (m - i) * a[m / 2];
        int j = upper_bound(a.begin() + m, a.end(), a[m + (n - m) / 2]) - a.begin();
        sum += (j - m) * a[m + (n - m) / 2] - (acc[j] - acc[m]);
        sum += (acc[n] - acc[j]) - (n - j) * a[m + (n - m) / 2];
        chmin(ans, sum);
    }
    return ans;
}

int main() {
    int n; scanf("%d", &n);
    vector<ll> a(n);
    REP (i, n) {
        scanf("%lld", &a[i]);
    }
    printf("%lld\n", solve(n, a));
    return 0;
}
0