結果

問題 No.1538 引きこもりさんは引き算が得意。
ユーザー AEn
提出日時 2022-11-20 17:02:03
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,081 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,200 KB
実行使用メモリ 86,452 KB
最終ジャッジ日時 2024-09-21 16:10:25
合計ジャッジ時間 5,401 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 50 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
N, K = map(int, input().split())
A = list(map(int, input().split()))

p, q = A[:N//2],A[N//2:]
P = [set() for _ in range(3)]
Q = [set() for _ in range(3)]

def cal(p, P):
    for i in range(1,1<<len(p)):
        s = []
        for j in range(len(p)):
            if i&(1<<j):
                s.append(p[j])
        for j in range(1<<len(s)):
            sm = 0
            cp, cn = 0,0
            for k in range(len(s)):
                if j&(1<<k):
                    cp += 1
                    sm += s[k]
                else:
                    cn += 1
                    sm -= s[k]
            if cp>=1 and cn>=1:
                P[0].add(sm)
            elif cp>=1:
                P[1].add(sm)
            else:
                P[2].add(sm)

cal(p,P)
cal(q,Q)
for n1 in P[0]:
    for i in range(3):
        if K-n1 in Q[i]:
            exit(print('Yes'))
for n1 in P[1]:
    if (K-n1 in Q[0]) or (K-n1 in Q[2]):
        exit(print('Yes'))
for n1 in P[2]:
    if (K-n1 in Q[0]) or (K-n1 in Q[1]):
        exit(print('Yes'))
print('No')
0