結果

問題 No.1017 Reiwa Sequence
ユーザー ygd.
提出日時 2021-05-18 15:00:35
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,016 bytes
コンパイル時間 529 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 309,348 KB
最終ジャッジ日時 2024-10-08 15:00:44
合計ジャッジ時間 38,968 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 3
other AC * 13 WA * 22 TLE * 10 -- * 5
権限があれば一括ダウンロードができます

ソースコード

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 dic.keys():
    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] = 1
                if (b>>s)&1 == 1:
                    ret[s] = -1
                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