結果

問題 No.1538 引きこもりさんは引き算が得意。
コンテスト
ユーザー shotoyoo
提出日時 2021-06-17 00:34:42
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 979 bytes
コンパイル時間 1,768 ms
コンパイル使用メモリ 81,900 KB
実行使用メモリ 114,956 KB
最終ジャッジ日時 2024-12-31 16:09:42
合計ジャッジ時間 7,113 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 52 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))


n,K = list(map(int, input().split()))
a = list(map(int, input().split()))
a0 = a[:n//2]
a1 = a[n//2:]
l0 = {}
l1 = {}
def sub(a):
    res = {}
    for b in range(pow(3,len(a))):
        val = 0
        p = m = 0
        for i in range(len(a)):
            if b%3==0:
                val += a[i]
                p += 1
            elif b%3==1:
                val -= a[i]
                m += 1
            b //= 3
        res.setdefault(val, set())
        vv = (p>0) | (2*(m>0))
        res[val].add(vv)
    return res
l0 = sub(a0)
l1 = sub(a1)
for k,v in l0.items():
    if K-k in l1:
        ss = v|l1[K-k]
        if 3 in ss or (1 in v and 2 in l1[K-k]) or (2 in v and 1 in l1[K-k]):
            print("Yes")
            break
else:
    print("No")
0