結果

問題 No.972 選び方のスコア
ユーザー kimiyukikimiyuki
提出日時 2020-01-18 02:54:19
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 125 ms / 2,000 ms
コード長 1,225 bytes
コンパイル時間 2,414 ms
コンパイル使用メモリ 207,168 KB
実行使用メモリ 8,064 KB
最終ジャッジ日時 2024-06-26 06:36:06
合計ジャッジ時間 6,574 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 124 ms
7,936 KB
testcase_01 AC 125 ms
7,808 KB
testcase_02 AC 118 ms
7,680 KB
testcase_03 AC 58 ms
5,504 KB
testcase_04 AC 115 ms
7,936 KB
testcase_05 AC 115 ms
7,936 KB
testcase_06 AC 115 ms
7,808 KB
testcase_07 AC 94 ms
6,656 KB
testcase_08 AC 106 ms
7,936 KB
testcase_09 AC 103 ms
7,680 KB
testcase_10 AC 106 ms
7,808 KB
testcase_11 AC 112 ms
7,936 KB
testcase_12 AC 112 ms
7,936 KB
testcase_13 AC 112 ms
7,936 KB
testcase_14 AC 112 ms
7,936 KB
testcase_15 AC 112 ms
7,936 KB
testcase_16 AC 113 ms
7,936 KB
testcase_17 AC 114 ms
7,808 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 110 ms
7,936 KB
testcase_20 AC 110 ms
7,936 KB
testcase_21 AC 111 ms
7,808 KB
testcase_22 AC 112 ms
7,680 KB
testcase_23 AC 111 ms
7,936 KB
testcase_24 AC 112 ms
8,064 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 2 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i))
#define ALL(x) std::begin(x), std::end(x)
using namespace std;
template <class T, class U> inline void chmax(T & a, U const & b) { a = max<T>(a, b); }
template <typename T> istream & operator >> (istream & in, vector<T> & xs) { REP (i, xs.size()) { in >> xs[i]; } return in; }

template <typename UnaryPredicate>
int64_t binsearch(int64_t l, int64_t r, UnaryPredicate p) {
    assert (l <= r);
    -- l;
    while (r - l > 1) {
        int64_t m = l + (r - l) / 2;
        (p(m) ? r : l) = m;
    }
    return r;
}

int64_t solve(int n, vector<int64_t> a) {
    sort(ALL(a));
    vector<int64_t> acc(n + 1);
    partial_sum(ALL(a), acc.begin() + 1);

    int64_t ans = INT64_MIN;
    REP (i, n) {
        int k_max = min(i, n - i - 1);
        auto f = [&](int k) {
            return a[i] + (acc[i] - acc[i - k]) + (acc[n] - acc[n - k]) - (2 * k + 1) * a[i];
        };
        int k = binsearch(0, k_max, [&](int k) {
            return f(k + 1) - f(k) < 0;
        });
        chmax(ans, f(k));
    }
    return ans;
}

int main() {
    int n; cin >> n;
    vector<int64_t> a(n); cin >> a;
    cout << solve(n, a) << endl;
    return 0;
}
0