結果

問題 No.1823 Tricolor Dango
ユーザー srjywrdnprkt
提出日時 2023-05-24 22:49:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 979 bytes
コンパイル時間 1,331 ms
コンパイル使用メモリ 109,680 KB
最終ジャッジ日時 2025-02-13 04:42:45
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 9 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <cmath>
#include <map>
#include <set>
#include <iomanip>
#include <queue>
#include <algorithm>
#include <numeric>
#include <deque>
#include <complex>
#include <cassert>

using namespace std;
using ll = long long;

void solve(){

    ll N, S=0, A, a, b, c;
    cin >> N;
    multiset<ll> st;;
    for (int i=0; i<N; i++){
        cin >> A;
        S += A;
        st.insert(A);
    }

    if (S % 3 != 0){
        cout << "No" << endl;
        return;
    }

    while(st.size() >= 3){
        a = *st.rbegin();
        st.erase(prev(st.end()));
        b = *st.rbegin();
        st.erase(prev(st.end()));
        c = *st.rbegin();
        st.erase(prev(st.end()));
        b -= c;
        if (b > 0) st.insert(b);
        a -= c;
        if (a > 0) st.insert(a);
    }

    cout << (st.size() == 0 ? "Yes" : "No") << endl;
}

int main(){

    int T;
    cin >> T;

    while(T){
        solve();
        T--;
    }

    return 0;
}
0