結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー maspy
提出日時 2020-05-14 18:51:34
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 921 bytes
コンパイル時間 676 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 29,672 KB
最終ジャッジ日時 2024-09-15 13:13:09
合計ジャッジ時間 9,126 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import bisect

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

N, P, Q = map(int, readline().split())
A = [0] + list(map(int, read().split()))

def next_perm(A):
    B = [A.pop()]
    while A[-1] > B[-1]:
        B.append(A.pop())
    x = A.pop()
    i = bisect.bisect_left(B, x)
    A.append(B[i])
    B[i] = x
    A += B
    return A

def solve(A, P, Q):
    next_perm(A)
    if A[0]:
        return False
    p = A.index(P)
    q = A.index(Q)
    if p < q:
        return A
    if max(A[q:]) == Q:
        while A[q] < A[q - 1]:
            q -= 1
    # とりあえず、Qを動かす
    A = A[:q + 1] + sorted(A[q + 1:], reverse=True)
    next_perm(A)
    p = A.index(P)
    q = A.index(Q)
    if p < q:
        return A
    del A[q]
    A.insert(p, Q)
    return A

A = solve(A, P, Q)
if not A or A[0] != 0:
    print(-1)
else:
    print(*A[1:])
0