結果

問題 No.1051 PQ Permutation
ユーザー roaris
提出日時 2020-05-11 18:34:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 185 ms / 2,000 ms
コード長 1,803 bytes
コンパイル時間 203 ms
コンパイル使用メモリ 82,536 KB
実行使用メモリ 164,740 KB
最終ジャッジ日時 2024-07-18 17:30:10
合計ジャッジ時間 7,915 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N, P, Q = map(int, input().split())
A = list(map(int, input().split()))
max1 = [-1]*(N+1)
max2 = [-2]*(N+1)

for i in range(N-1, -1, -1):
    if A[i]>max1[i+1]:
        max1[i] = A[i]
        max2[i] = max1[i+1]
    elif A[i]>max2[i+1]:
        max1[i] = max1[i+1]
        max2[i] = A[i]
    else:
        max1[i] = max1[i+1]
        max2[i] = max2[i+1]
        
match = -1
s = set()

for i in range(N):
    if (P not in s) and (Q not in s):
        if max1[i]!=Q and max1[i]>A[i]:
            match = i
        elif max1[i]==Q and max2[i]>A[i]:
            match = i
    elif (P not in s) and (Q in s):
        break
    else:
        if max1[i]>A[i]:
            match = i

    s.add(A[i])

if match==-1:
    print(-1)
    exit()

ans = []
rem = []

for i in range(N):
    if i<match:
        ans.append(A[i])
    else:
        rem.append(A[i])

s = set(ans)
rem.sort()

if (P not in s) and (Q not in s):
    for i in range(len(rem)):
        if rem[i]==A[match]:
            if rem[i+1]!=Q:
                mark = i+1
            else:
                mark = i+2
    
    ans.append(rem[mark])
    rem2 = [rem[i] for i in range(len(rem)) if i!=mark]
    
    if P<Q:
        ans += rem2
    else:
        if rem[mark]==P:
            ans += rem2
        else:
            for rem2_i in rem2:
                if rem2_i<P and rem2_i!=Q:
                    ans.append(rem2_i)
        
            ans.append(P)
            ans.append(Q)
        
            for rem2_i in rem2:
                if rem2_i>P:
                    ans.append(rem2_i)
else:
    for i in range(len(rem)):
        if rem[i]==A[match]:
            mark = i+1
    
    ans.append(rem[mark])
    
    for i in range(len(rem)):
        if i!=mark:
            ans.append(rem[i])

print(*ans)
0