結果

問題 No.1051 PQ Permutation
ユーザー kriiikriii
提出日時 2020-05-10 08:29:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 1,107 bytes
コンパイル時間 473 ms
コンパイル使用メモリ 46,388 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-07 03:32:30
合計ジャッジ時間 3,397 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <algorithm>
using namespace std;

int N, P, Q, A[200200], p, q;

int main()
{
	scanf ("%d %d %d", &N, &P, &Q);
	for (int i = 0; i < N; i++) scanf ("%d", &A[i]);

	if (!next_permutation(A, A + N)){
		puts("-1");
		return 0;
	}

	for (int i = 0; i < N; i++){
		if (A[i] == P) p = i;
		if (A[i] == Q) q = i;
	}

	if (p < q){
		for (int i = 0; i < N; i++) printf ("%d ", A[i]);
		return 0;
	}

	int u = 0;
	for (int i = q + 1; i < N; i++) u = max(u, A[i]);
	if (Q > u){
		int w = Q;
		for (q--; q >= 0; q--){
			if (w < A[q]) w = A[q];
			else if (u < A[q] && A[q] < Q) u = A[q];
			else break;
		}
	}
	sort(A + q + 1, A + N);
	reverse(A + q + 1, A + N);

	if (!next_permutation(A, A + N)){
		puts("-1");
		return 0;
	}

	for (int i = 0; i < N; i++){
		if (A[i] == P) p = i;
		if (A[i] == Q) q = i;
	}

	if (p < q){
		for (int i = 0; i < N; i++) printf ("%d ", A[i]);
		return 0;
	}

	for (int i = 0; i < q; i++) printf ("%d ", A[i]);
	for (int i = q + 1; i < p; i++) printf ("%d ", A[i]);
	printf ("%d %d ", P, Q);
	for (int i = p + 1; i < N; i++) printf ("%d ", A[i]);
	return 0;
}
0