結果

問題 No.1594 Three Classes
コンテスト
ユーザー Kohei
提出日時 2026-07-28 22:29:23
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 748 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 237 ms
コンパイル使用メモリ 95,720 KB
実行使用メモリ 87,808 KB
最終ジャッジ日時 2026-07-28 22:29:53
合計ジャッジ時間 6,554 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
sys.setrecursionlimit(10**8)
import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
N = int(input())
E = list(map(int, input().split()))
sumE = sum(E)
if sumE%3 != 0:
    exit(print('No'))
oneclass = sumE//3

ans = 'No'
def dfs(done:list, team, total):
    global ans
    if team == 2:
        ans = 'Yes'
        return

    for i in range(N):
        if done[i] == 0:
            if total + E[i] == oneclass:
                done[i] = 1
                dfs(done, team + 1, 0)
                done[i] = 0
            elif total + E[i] < oneclass:
                done[i] = 1
                dfs(done, team, total + E[i])
                done[i] = 0
            # オーバーするなら遷移しない

dfs([0]*N, 0, 0)
print(ans)
0