結果

問題 No.1017 Reiwa Sequence
ユーザー ygd.
提出日時 2021-05-18 15:06:44
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,030 bytes
コンパイル時間 340 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 855,604 KB
最終ジャッジ日時 2024-10-08 15:10:03
合計ジャッジ時間 9,228 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 MLE * 1 -- * 2
other -- * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
A = list(map(int,input().split()))

num = min(N, 22)

from collections import defaultdict 
dic = defaultdict(list)

for i in range(1,1<<num):
    temp = 0
    for j in range(num):
        if (i>>j)&1 == 1:
            temp += A[j]
    dic[temp].append(i)
#print(dic)

def overlap(p,q):
    for i in range(30):
        if (p>>i)&1 == 1 and (q>>i)&1 == 1:
            return False
    return True

ret = [0]*N

for x in range(2*pow(10,7)):
    if len(dic[x]) <= 1: continue
    n = len(dic[x])
    for i in range(n):
        a = dic[x][i]
        for j in range(i+1,n):
            b = dic[x][j]
            if not overlap(a,b): continue
            for s in range(30):
                if (a>>s)&1 == 1:
                    ret[s] = A[s]
                if (b>>s)&1 == 1:
                    ret[s] = -A[s]
                Flag = 0
                if (a>>s)&1 == 1 and (b>>s)&1 == 1:
                    Flag = 1
                assert Flag == 0
            print("Yes")
            print(*ret); exit()
print("No")
0