結果

問題 No.1906 Twinkle Town
コンテスト
ユーザー Sumitacchan
提出日時 2022-02-11 15:57:46
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 3,454 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,447 ms
コンパイル使用メモリ 203,628 KB
最終ジャッジ日時 2025-01-27 21:14:47
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 47
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
//using namespace atcoder;
struct fast_ios { fast_ios(){ cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_;
#define FOR(i, begin, end) for(int i=(begin);i<(end);i++)
#define REP(i, n) FOR(i,0,n)
#define IFOR(i, begin, end) for(int i=(end)-1;i>=(begin);i--)
#define IREP(i, n) IFOR(i,0,n)
#define Sort(v) sort(v.begin(), v.end())
#define Reverse(v) reverse(v.begin(), v.end())
#define all(v) v.begin(),v.end()
#define SZ(v) ((int)v.size())
#define Lower_bound(v, x) distance(v.begin(), lower_bound(v.begin(), v.end(), x))
#define Upper_bound(v, x) distance(v.begin(), upper_bound(v.begin(), v.end(), x))
#define chmax(a, b) a = max(a, b)
#define chmin(a, b) a = min(a, b)
#define bit(n) (1LL<<(n))
#define debug(x) cout << #x << "=" << x << endl;
#define vdebug(v) { cout << #v << "=" << endl; REP(i_debug, (int)v.size()){ cout << v[i_debug] << ","; } cout << endl; }
#define mdebug(m) { cout << #m << "=" << endl; REP(i_debug, (int)m.size()){ REP(j_debug, (int)m[i_debug].size()){ cout << m[i_debug][j_debug] << ","; } cout << endl;} }
#define pb push_back
#define fi first
#define se second
#define int long long
#define INF 1000000000000000000
template<typename T> istream &operator>>(istream &is, vector<T> &v){ for (auto &x : v) is >> x; return is; }
template<typename T> ostream &operator<<(ostream &os, vector<T> &v){ for(int i = 0; i < (int)v.size(); i++) { cout << v[i]; if(i != (int)v.size() - 1) cout << endl; }; return os; }
template<typename T1, typename T2> ostream &operator<<(ostream &os, pair<T1, T2> p){ cout << '(' << p.first << ',' << p.second << ')'; return os; }
template<typename T> void Out(T x) { cout << x << endl; }
template<typename T1, typename T2> void chOut(bool f, T1 y, T2 n) { if(f) Out(y); else Out(n); }

using vec = vector<int>;
using mat = vector<vec>;
using Pii = pair<int, int>;
using v_bool = vector<bool>;
using v_Pii = vector<Pii>;

//int dx[4] = {1,0,-1,0};
//int dy[4] = {0,1,0,-1};
//char d[4] = {'D','R','U','L'};

//const int mod = 1000000007;
const int mod = 998244353;

int solve_even(vec A, int N){
    int ans = 0, t = 0;
    // 右から何個とるか
    REP(k, N / 2 + 1){
        if(k >= 1){
            int i = N - 2 - 2 * (k - 1);
            t += (i > 0 ? A[i] - A[i - 1] : A[i]);
        }
        
        int rest = N / 2 - k;
        chmax(ans, t + (rest > 0 ? A[rest - 1] : 0));
    }
    return ans;
}

int solve_odd(vec A, int N){
    if(N == 1) return A[0];
    int ans = 0;

    // 最初にA[N-1]
    chmax(ans, A[N - 1] - A[N - 2] + solve_even(A, N - 1));

    // 最初にそれ以外
    mat dp(N + 1, vec(2, -INF));
    dp[N][0] = 0;
    IFOR(i, 1, N){
        if(i + 2 <= N){
            REP(j, 2) chmax(dp[i][j], dp[i + 2][j] + A[i] - A[i - 1]);
        }
        if(i + 3 <= N){
            chmax(dp[i][1], dp[i + 3][0] + A[i] - A[i - 1]);
        }
    }
    FOR(i, 1, N + 1){
        int rest = N / 2 - (N - i) / 2;
        if(rest == 0){
            REP(j, 2) chmax(ans, dp[i][j] + A[i - 1]);
        }else{
            REP(j, 2) chmax(ans, dp[i][j] + A[rest - 1 + j]);
        }
    }

    return ans;
}

signed main(){

    int T; cin >> T;
    vec ans(T);
    REP(q, T){
        int N; cin >> N;
        vec A(N); cin >> A;
        Sort(A);

        if(N % 2) ans[q] = solve_odd(A, N);
        else ans[q] = A.back() - solve_even(A, N);
    }
    cout << ans << endl;

    return 0;
}
0