結果

問題 No.1373 Directed Operations
コンテスト
ユーザー qwewe
提出日時 2025-05-14 12:47:24
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,064 bytes
コンパイル時間 434 ms
コンパイル使用メモリ 82,724 KB
実行使用メモリ 108,284 KB
最終ジャッジ日時 2025-05-14 12:48:47
合計ジャッジ時間 3,478 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 9 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    input = sys.stdin.read().split()
    N = int(input[0])
    a_list = list(map(int, input[1:N]))
    
    reachable = {1}
    SMN = 2
    rightmost = 1
    result = []
    
    for a in a_list:
        # Try to add SMN
        if (SMN - a) in reachable and (SMN - a) >= 1:
            x = SMN - a
            result.append(x)
            reachable.add(SMN)
            # Update SMN
            while SMN <= N and SMN in reachable:
                SMN += 1
        else:
            # Try to add rightmost + a
            new_node = rightmost + a
            if new_node > N or new_node in reachable:
                print("NO")
                return
            result.append(rightmost)
            reachable.add(new_node)
            rightmost = new_node
            # Check if new_node is SMN
            while SMN <= N and SMN in reachable:
                SMN += 1
    
    if len(reachable) == N:
        print("YES")
        print('\n'.join(map(str, result)))
    else:
        print("NO")

if __name__ == "__main__":
    main()
0