#480005 (C++17) No.1051 PQ Permutation

• 問題: No.1051 PQ Permutation
• ユーザー: kriii
• 提出日時: 2020-05-10 08:51:49
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 81 ms / 2,000 ms
• コード長: 1,032 bytes
• コンパイル時間: 680 ms
• コンパイル使用メモリ: 74,132 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-07 04:04:45
• 合計ジャッジ時間: 4,580 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	80 ms; 5,376 KB
testcase_16	AC	81 ms; 5,376 KB
testcase_17	AC	80 ms; 5,376 KB
testcase_18	AC	80 ms; 5,376 KB
testcase_19	AC	80 ms; 5,376 KB
testcase_20	AC	81 ms; 5,376 KB
testcase_21	AC	81 ms; 5,376 KB
testcase_22	AC	80 ms; 5,376 KB
testcase_23	AC	81 ms; 5,376 KB
testcase_24	AC	81 ms; 5,376 KB
testcase_25	AC	8 ms; 5,376 KB
testcase_26	AC	5 ms; 5,376 KB
testcase_27	AC	3 ms; 5,376 KB
testcase_28	AC	12 ms; 5,376 KB
testcase_29	AC	5 ms; 5,376 KB
testcase_30	AC	8 ms; 5,376 KB
testcase_31	AC	4 ms; 5,376 KB
testcase_32	AC	7 ms; 5,376 KB
testcase_33	AC	3 ms; 5,376 KB
testcase_34	AC	3 ms; 5,376 KB
testcase_35	AC	39 ms; 5,376 KB
testcase_36	AC	40 ms; 5,376 KB
testcase_37	AC	61 ms; 5,376 KB
testcase_38	AC	81 ms; 5,376 KB
testcase_39	AC	62 ms; 5,376 KB
testcase_40	AC	79 ms; 5,376 KB
testcase_41	AC	80 ms; 5,376 KB
testcase_42	AC	2 ms; 5,376 KB
testcase_43	AC	2 ms; 5,376 KB
testcase_44	AC	2 ms; 5,376 KB
testcase_45	AC	2 ms; 5,376 KB
testcase_46	AC	2 ms; 5,376 KB
testcase_47	AC	2 ms; 5,376 KB
testcase_48	AC	62 ms; 5,376 KB

ソースコード

#include <iostream>
#include <algorithm>
using namespace std;

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

int main()
{
	cin >> N >> P >> Q;
	for (int i = 0; i < N; i++) cin >> A[i];

	for (int k = 0; next_permutation(A, A + N); k++){
		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;
		}

		if (k){
			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;
		}
		else{
			int u = 0, w = Q;
			for (int i = q + 1; i < N; i++) u = max(u, A[i]);
			if (Q > u){
				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;
				}
			}

			for (int i = q + 1; i < N; i++) C[A[i]] = 1;
			for (int i = 1, j = q; i <= N; i++) if (C[i]) A[++j] = i;
			reverse(A + q + 1, A + N);
		}
	}

	puts("-1");
	return 0;
}