結果

問題 No.972 選び方のスコア
コンテスト
ユーザー kimiyuki
提出日時 2020-01-18 02:12:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 906 bytes
コンパイル時間 2,644 ms
コンパイル使用メモリ 200,092 KB
最終ジャッジ日時 2025-01-08 19:49:13
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 7
権限があれば一括ダウンロードができます

ソースコード

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; }

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 + 1) / 2) {
        chmax(ans, a[i] + (acc[i] - acc[0]) + (acc[n] - acc[n - i]) - (2 * i + 1) * a[i]);
        chmax(ans, a[i] + a[i + 1] + (acc[i] - acc[0]) + (acc[n] - acc[n - i]) - (2 * i + 2) * (a[i] + a[i + 1]) / 2);
    }
    return ans;
}

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