結果

問題 No.1594 Three Classes
コンテスト
ユーザー norioc
提出日時 2021-07-10 05:05:58
言語 Python3
(3.14.3 + numpy 2.4.2 + scipy 1.17.0)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
AC  
実行時間 1,511 ms / 2,000 ms
コード長 434 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 588 ms
コンパイル使用メモリ 20,700 KB
実行使用メモリ 15,484 KB
最終ジャッジ日時 2026-03-23 04:26:08
合計ジャッジ時間 16,515 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def bits(p, n):
    a = [0] * n
    for i in range(pow(p, n)):
        x = i
        for j in range(n):
            a[j] = x % p
            x //= p
        yield a

input()
a = list(map(int, input().split()))

ok = False
b = [0, 0, 0]
for es in bits(3, len(a)):
    b[0] = b[1] = b[2] = 0
    for i in range(len(a)):
        b[es[i]] += a[i]
    if b[0] == b[1] == b[2]:
        ok = True
        break

print('Yes' if ok else 'No')
0