結果

問題 No.1594 Three Classes
コンテスト
ユーザー Tsukumowan
提出日時 2021-07-10 13:52:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 104 ms / 2,000 ms
コード長 929 bytes
コンパイル時間 2,592 ms
コンパイル使用メモリ 194,296 KB
最終ジャッジ日時 2025-01-23 00:00:53
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

int power(int p){
    int res = 1;
    for(int i = 0;i < p;i++) res *= 3;
    return res;
}

int main(){
    int n;
    cin >> n;
    vector<int64_t> v(n);
    int64_t sum = 0;
    for(int i = 0;i < n;i++){
        cin >> v[i];
        sum += v[i];
    }
    bool ans = false;
    if(sum % 3 == 0){
        for(int i = 0;i < power(n);i++){
            int64_t a = 0;
            int64_t b = 0;
            int64_t c = 0;
            int bi = i;
            for(int j = n;j > 0;j--){
                int rem = bi / power(j);
                if(rem == 0) a += v[j - 1];
                else if(rem == 1) b += v[j - 1];
                else c += v[j - 1];
                bi %= power(j);
            }
            if(a == b && b == c){
                ans = true;
                break;
            }
        }
    }
    if(ans) cout << "Yes" << endl;
    else cout << "No" << endl;
}
0