#675327 (Python3) No.1594 Three Classes

• 問題: No.1594 Three Classes
• ユーザー: tktk_snsn
• 提出日時: 2021-07-09 21:28:04
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 1,657 ms / 2,000 ms
• コード長: 284 bytes
• コンパイル時間: 66 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 10,752 KB
• 最終ジャッジ日時: 2024-04-27 23:55:55
• 合計ジャッジ時間: 15,726 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1,586 ms; 10,752 KB
testcase_01	AC	26 ms; 10,624 KB
testcase_02	AC	25 ms; 10,752 KB
testcase_03	AC	86 ms; 10,624 KB
testcase_04	AC	1,604 ms; 10,496 KB
testcase_05	AC	1,593 ms; 10,752 KB
testcase_06	AC	27 ms; 10,624 KB
testcase_07	AC	58 ms; 10,624 KB
testcase_08	AC	1,518 ms; 10,624 KB
testcase_09	AC	1,515 ms; 10,752 KB
testcase_10	AC	1,657 ms; 10,752 KB
testcase_11	AC	1,523 ms; 10,496 KB
testcase_12	AC	1,489 ms; 10,752 KB
testcase_13	AC	26 ms; 10,624 KB
testcase_14	AC	223 ms; 10,624 KB
testcase_15	AC	207 ms; 10,752 KB
testcase_16	AC	105 ms; 10,752 KB
testcase_17	AC	80 ms; 10,496 KB
testcase_18	AC	24 ms; 10,624 KB
testcase_19	AC	28 ms; 10,624 KB
testcase_20	AC	35 ms; 10,624 KB

ソースコード

from itertools import product
N = int(input())
A = list(map(int, input().split()))

for pair in product(range(3), repeat=N):
    sum = [0, 0, 0]
    for i, p in enumerate(pair):
        sum[p] += A[i]
    if sum[0] == sum[1] == sum[2]:
        print("Yes")
        exit()
print("No")