結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー tac
提出日時 2020-05-09 18:26:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,703 bytes
コンパイル時間 3,221 ms
コンパイル使用メモリ 213,768 KB
最終ジャッジ日時 2025-01-10 09:59:16
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 2 WA * 42 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define F first
#define S second
#define pii pair<int, int>
#define eb emplace_back
#define all(v) v.begin(), v.end()
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep3(i, l, n) for (int i = l; i < (n); ++i)
#define sz(v) (int)v.size()
#define endl '\n'
const int inf = 1000000007;
const ll INF = 1e18;
// int mod = 998244353;
int mod = 1000000007;
#define abs(x) (x >= 0 ? x : -(x))
#define lb(v, x) (int)(lower_bound(all(v), x) - v.begin())
#define ub(v, x) (int)(upper_bound(all(v), x) - v.begin())
template<typename T1, typename T2> inline bool chmin(T1 &a, T2 b) { if (a > b) { a = b; return 1; } return 0; }
template<typename T1, typename T2> inline bool chmax(T1 &a, T2 b) { if (a < b) { a = b; return 1; } return 0; }
ll gcd(ll a, ll b) { if (b == 0) return a; return gcd(b, a % b); }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
template<typename T, typename U> T pow_(T a, U b) { return b ? pow_(a * a, b / 2) * (b % 2 ? a : 1) : 1; }
ll modpow(ll a, ll b, ll _mod) { return b ? modpow(a * a % _mod, b / 2, _mod) * (b % 2 ? a : 1) % _mod : 1; }
template<class T, class U> ostream& operator << (ostream& os, const pair<T, U>& p) { os << p.F << " " << p.S; return os; }
template<class T> ostream& operator << (ostream& os, const vector<T>& vec) { rep(i, sz(vec)) { if (i) os << " "; os << vec[i]; } return os; }
template<typename T> inline istream& operator >> (istream& is, vector<T>& v) { rep(j, sz(v)) is >> v[j]; return is; }
template<class T, class T2> inline void add(T &a, T2 b) { a += b; if (a >= mod) a -= mod; }


void solve();

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cout << fixed << setprecision(10);
    int T;
    T = 1;
    while (T--) solve();
}

void solve() {
    int n, p, q;
    cin >> n >> p >> q;
    vector<int> a(n);
    cin >> a;
    rep(i, n) { // corner, next_permutation 1 回で最小のになる
        if (a[i] != n - i) break;
        if (i == n - 1) {
            if (p > q) cout << a << endl;
            else cout << -1 << endl;
            return;
        }
    }
    
    int c = 0;
    do {
        if (c == 1) break;
        c++;
    } while (next_permutation(all(a)));
    cout << a << endl;
    
    map<int, int> mp;
    rep(i, n) mp[a[i]] = i;
    if (mp[p] < mp[q]) {
        cout << a << endl;
        return;
    }
    if (q == n && mp[q] == 0) { // sample 3
        cout << -1 << endl;
        return;
    }
    
    vector<int> b;
    set<int> s;
    rep3(i, 1, n + 1) s.insert(i);
    
    // もう p に入った, 自由
    auto dfs = [&](auto&& dfs, int p_, int fr) -> void {
        // cout << b << "  " << p_ << " " << fr << endl;
        if (fr) {
            vector<int> c;
            if (p_) {
                for (int e : s) c.eb(e);
                sort(all(c));
            } else {
                for (int e : s) if (e != p && e != q) c.eb(e);
                sort(all(c));
                int it = lb(c, p);
                c.insert(c.begin() + it, p);
                it = lower_bound(c.begin() + it + 1, c.end(), q) - c.begin();
                c.insert(c.begin() + it, q);
            }
            rep(i, sz(c)) b.eb(c[i]);
            // cout << b << endl;
            exit(0);
        } else {
            rep3(i, a[sz(b)], n + 1) {
                if (p_ == 0 && i == q) continue;
                if (!s.count(i)) continue;
                
                b.eb(i);
                s.erase(i);
                dfs(dfs, p_ | (i == p), fr | (i > a[sz(b) - 1]));
                b.pop_back();
                s.insert(i);
            }
        }
    };
    dfs(dfs, 0, 0);
    cout << -1 << endl;
}
0