結果

問題 No.1051 PQ Permutation
ユーザー roaris
提出日時 2020-05-15 02:35:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 218 ms / 2,000 ms
コード長 2,080 bytes
コンパイル時間 245 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 164,448 KB
最終ジャッジ日時 2024-09-16 16:12:41
合計ジャッジ時間 7,649 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 (Q in s) and (P not in s):
        break
    
    if (P not in s) and (Q not in s):
        if max1[i]!=Q and A[i]<max1[i]:
            match = i
        elif A[i]<max2[i]:
            match = i
    else:
        if A[i]<max1[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])

rem.sort()
s = set(ans)

if P in s:
    rem2 = []
    
    for i in range(len(rem)):
        if rem[i]==A[match]:
            mark = i+1
            
            for j in range(len(rem)):
                if j!=mark:
                    rem2.append(rem[j])
                    
            break
    
    ans.append(rem[mark])
    ans += rem2
else:
    rem2 = []
    
    for i in range(len(rem)):
        if rem[i]==A[match]:
            if rem[i+1]!=Q:
                mark = i+1
            else:
                mark = i+2
            
            for j in range(len(rem)):
                if j!=mark:
                    rem2.append(rem[j])
            
            break
    
    if P<Q:
        ans.append(rem[mark])
        ans += rem2
    else:
        if rem[mark]==P:
            ans.append(rem[mark])
            ans += rem2
        else:
            ans.append(rem[mark])
        
            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)
    
print(*ans)
0