#478875 (C++11) No.1051 PQ Permutation

• 問題: No.1051 PQ Permutation
• ユーザー: leaf_1415
• 提出日時: 2020-05-08 22:26:14
• 言語: C++11 (gcc 11.4.0)
• 結果: WA
• 実行時間: -
• コード長: 2,245 bytes
• コンパイル時間: 1,200 ms
• コンパイル使用メモリ: 82,900 KB
• 実行使用メモリ: 22,596 KB
• 最終ジャッジ日時: 2023-09-19 07:45:10
• 合計ジャッジ時間: 6,945 ms
• ジャッジサーバーID （参考情報）: judge11 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	WA	-
testcase_02	AC	1 ms; 4,380 KB
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-
testcase_36	WA	-
testcase_37	WA	-
testcase_38	WA	-
testcase_39	WA	-
testcase_40	WA	-
testcase_41	WA	-
testcase_42	WA	-
testcase_43	WA	-
testcase_44	WA	-
testcase_45	WA	-
testcase_46	WA	-
testcase_47	WA	-
testcase_48	WA	-

ソースコード

#include <iostream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstdlib>
#include <cassert>
#include <vector>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <bitset>
#include <string>
#include <algorithm>
#include <utility>
#define llint long long
#define inf 1e18
#define rep(x, s, t) for(llint (x) = (s); (x) < (t); (x)++)
#define Rep(x, s, t) for(llint (x) = (s); (x) <= (t); (x)++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))
#define mod 1000000007

using namespace std;
typedef pair<llint, llint> P;

llint n, p, q;
llint a[200005], b[200005];
bool ok[200005];

set<llint> S;
bool dfs(llint i)
{
	if(i == n) return false;
	
	bool ret = false;
	if(S.count(p) == 0){
		bool flag = false;
		for(int j = i; j < n; j++){
			if(a[j] < a[j+1]) flag = true;
		}
		ret |= flag;
	}
	else{
		if(a[i] != q){
			S.insert(a[i]);
			ret |= dfs(i+1);
			S.erase(a[i]);
		}
	}
	
	auto it = S.upper_bound(a[i]);
	if(!(it == S.end() || (*it == q && S.count(p)))) ret = true;
	it++;
	if(!(it == S.end() || (*it == q && S.count(p)))) ret = true;
	
	return ok[i] = ret;
}

void make(llint l)
{
	cout << l << endl;
	for(int i = l; i <= n; i++){
	//for(auto x:S) cout << x << " "; cout << endl;
		
		auto it = S.begin();
		if(!(*it == q && S.count(p))){
			b[i] = *it;
			S.erase(*it);
			continue;
		}
		it++;
		if(!(*it == q && S.count(p))){
			b[i] = *it;
			S.erase(*it);
			continue;
		}
		
	}
}

int main(void)
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n >> p >> q;
	for(int i = 1; i <= n; i++) cin >> a[i];
	
	for(int i = 1; i <= n; i++) S.insert(i);
	dfs(1);
	//for(int i = 1; i <= n; i++) cout << ok[i] << " "; cout << endl;
	
	if(!ok[1]){
		cout << -1 << endl;
		return 0;
	}
	
	for(int i = 1; i <= n; i++){
		if(ok[i+1]){
			b[i] = a[i];
			S.erase(a[i]);
			continue;
		}
		auto it = S.upper_bound(a[i]);
		if(!(it == S.end() || (*it == q && S.count(p) == 0))){
			b[i] = *it;
			S.erase(*it);
			make(i+1);
			break;
		}
		it++;
		if(!(it == S.end() || (*it == q && S.count(p) == 0))){
			b[i] = *it;
			S.erase(*it);
			make(i+1);
			break;
		}
	}
	
	for(int i = 1; i <= n; i++) cout << b[i] << " "; cout << endl;
	
	return 0;
}