結果

問題 No.1594 Three Classes
コンテスト
ユーザー qqq xxa
提出日時 2021-07-18 17:34:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 850 bytes
コンパイル時間 1,793 ms
コンパイル使用メモリ 168,912 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-16 08:29:27
合計ジャッジ時間 2,052 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

bool solve() {
    ll N;
    cin >> N;
    vector<ll> E(N);
    for ( int i = 0; i < N; i++ ) cin >> E[i];
    ll s = 0;
    for ( int i = 0; i < N; i++ ) {
        s += E[i];
    }
    if ( s % 3 != 0 ) return false;
    ll s3 = s/3;
    auto get_sum = [&](int b) -> ll {
        ll t = 0;
        for ( int i = 0; i < N; i++ ) {
            if ( (b>>i)&1 ) t += E[i];
        }
        return t;
    };
    for ( int b1 = 1; b1 < 1<<N; b1++ ) {
        if ( get_sum(b1) != s3 ) continue;
        int bm = ((1<<N)-1) ^ b1;
        for ( int b2 = bm; b2 > 0; b2=((b2-1)&bm) ) {
            if ( get_sum(b2) != s3 ) continue;
            return true;
        }
    }
    return false;
}

int main() {
    string ans = solve() ? "Yes" : "No";
    cout << ans << "\n";
    return 0;
}
0