結果

問題 No.1538 引きこもりさんは引き算が得意。
ユーザー rin204
提出日時 2022-03-24 14:58:40
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 816 bytes
コンパイル時間 471 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 92,800 KB
最終ジャッジ日時 2024-10-12 23:23:02
合計ジャッジ時間 4,513 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 48 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

def yes():
    print("Yes")
    exit()
    
def no():
    print("No")
    exit()

n, k = map(int, input().split())
A = list(map(int, input().split()))
if k in A:
    yes()
L = A[:n // 2]
R = A[n // 2:]

def f(A):
    pm = set()
    p = set()
    for a in A:
        add = {a}
        for s in p:
            add.add(s + a)
        
        add2 = set()
        for s in p:
            add2.add(abs(s - a))
        for s in pm:
            add2.add(abs(s + a))
            add2.add(abs(s - a))
            
        p |= add
        pm |= add2
    return p, pm
        
p1, pm1 = f(L)
p2, pm2 = f(R)

if k in pm1 or k in pm2:
    yes()

if -k in pm1 or -k in pm2:
    yes()

se1 = p1 | pm1
se2 = p2 | pm2
    
for s in se1:
    for x in [s + k, s - k, -s + k, -s - k]:
        if x in se2:
            yes()
no()
0