結果

問題 No.1538 引きこもりさんは引き算が得意。
ユーザー 👑 rin204
提出日時 2022-03-24 15:05:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 114 ms / 2,000 ms
コード長 956 bytes
コンパイル時間 816 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 91,072 KB
最終ジャッジ日時 2024-10-12 23:30:04
合計ジャッジ時間 5,161 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

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(s - a)
            add2.add(a - s)
        for s in pm:
            for x in [s + a, s - a, -s + a, -s - a]:
                add2.add(x)

        p |= add
        pm |= add2
    return p, pm
        
p1, pm1 = f(L)
p2, pm2 = f(R)

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

for s in pm1:
    for x in [s + k, s - k, -s + k, -s - k]:
        if x in p2 or x in pm2:
            yes()

for s in p1:
    for x in [s + k, s - k]:
        if x in p2 or x in pm2:
            yes()
    for x in [-s + k, -s - k]:
        if x in pm2:
            yes()
            
no()
0