結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー walkre
提出日時 2020-05-08 23:15:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 4,914 bytes
コンパイル時間 1,702 ms
コンパイル使用メモリ 178,228 KB
実行使用メモリ 15,744 KB
最終ジャッジ日時 2024-07-05 20:00:57
合計ジャッジ時間 8,046 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32 TLE * 1 -- * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using i64 = int64_t;

#define rep(i, x, y) for (i64 i = i64(x), i##_max_for_repmacro = i64(y); i < i##_max_for_repmacro; ++i)
#define debug(x) #x << "=" << (x)

#ifdef DEBUG
#define _GLIBCXX_DEBUG
#define print(x) std::cerr << debug(x) << " (L:" << __LINE__ << ")" << std::endl
#else
#define print(x)
#endif

template <i64 p>
class fp {
    public:
    i64 x;
    fp() : x(0) {}
    fp(i64 x_) : x((x_ % p + p) % p) {}
    fp operator+() const { return fp(x); }
    fp operator-() const { return fp(-x); }
    fp& operator+=(const fp& y) {
        x += y.x;
        if (x >= p) x -= p;
        return *this;
    }
    fp& operator-=(const fp& y) { return *this += -y; }
    fp& operator*=(const fp& y) {
        x = x * y.x % p;
        return *this;
    }
    fp& operator/=(const fp& y) { return *this *= fp(inverse(y.x)); }
    fp operator+(const fp& y) const { return fp(x) += y; }
    fp operator-(const fp& y) const { return fp(x) -= y; }
    fp operator*(const fp& y) const { return fp(x) *= y; }
    fp operator/(const fp& y) const { return fp(x) /= y; }
    bool operator==(const fp& y) const { return x == y.x; }
    bool operator!=(const fp& y) const { return !(*this == y); }
    i64 extgcd(i64 a, i64 b, i64& x, i64& y) {
        i64 d = a;
        if (b != 0) {
            d = extgcd(b, a % b, y, x);
            y -= (a / b) * x;
        } else {
            x = 1;
            y = 0;
        }
        return d;
    }
    i64 inverse(i64 a) {
        i64 x, y;
        extgcd(a, p, x, y);
        return (x % p + p) % p;
    }
};

template <i64 p>
i64 abs(const fp<p>& x) { return x.x; }

template <i64 p>
istream& operator>>(istream& is, fp<p>& x) {
    is >> x.x;
    return is;
}

template <i64 p>
ostream& operator<<(ostream& os, const fp<p>& x) {
    os << x.x;
    return os;
}

template <typename T, typename U>
ostream& operator<<(ostream& os, const pair<T, U>& p) {
    os << "(" << p.first << ", " << p.second << ")";
    return os;
}

template <typename T>
ostream& operator<<(ostream& os, const vector<T>& vec) {
    os << "[";
    for (const auto& v : vec) {
        os << v << ",";
    }
    os << "]";
    return os;
}

template <typename T>
bool chmin(T& a, const T& b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}

template <typename T>
bool chmax(T& a, const T& b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}

template <typename A, typename T, size_t size>
void fill(A (&ary)[size], const T& val) {
    fill((T*)ary, (T*)(ary + size), val);
}

constexpr int inf = 1.01e9;
constexpr i64 inf64 = 4.01e18;
constexpr long double eps = 1e-9;

// double(64bit浮動小数)のn分探索のループ回数の上限(2分探索なら50でも十分かもしれない). long double(80ビットの x87 浮動小数点型?)だと, 2分探索であってもこれだと足りないケースがある気がするので, もうちょっと余裕を持たせた方が良さそう.
constexpr i64 max_loop = 100;

void solve() {
    //constexpr i64 mod = 1'000'000'007;

    i64 N,P,Q;
    cin >> N >> P >> Q;

    --P;
    --Q;

    vector<i64> A(N);
    rep(i,0,N){
        cin >> A[i];
        --A[i];
    }

    /*
    i64 pi=find(begin(A),end(A),P)-begin(A);
    i64 qi=find(begin(A),end(A),Q)-begin(A);
    if(pi<qi){

        return;
    }

    priority_queue<i64,vector<i64>,greater<i64>> pending,candidate;
    rep(i,0,N){
        if(i==Q) continue;
        candidate.push(i);
    }

    bool over=false;
    vector<i64> ans(N);
    rep(i,0,N){
        bool ok=false;
        while(!pending.empty()){
            i64 x=pending.top();
            if(over or x>=A[i]){
                if(x>A[i]) over=true;
                ans[i]=x;
                ok=true;
                break;
            }
        }
    }
     */

    set<i64> st;
    rep(i,0,N) if(i!=Q) st.insert(i);

    bool over=false;
    vector<i64> ans(N);
    rep(i,0,N){
        auto it=over?st.begin():st.lower_bound(A[i]);
        if(it==end(st)){
            over=true;
            it=st.begin();
        }

        i64 x=*it;
        ans[i]=x;
        if(A[i]<x) over=true;
        st.erase(x);
        if(x==P) st.insert(Q);
    }

    if(ans<=A){
        bool ok=false;
        do{
            i64 pi=find(begin(ans),end(ans),P)-begin(ans);
            i64 qi=find(begin(ans),end(ans),Q)-begin(ans);
            if(pi<qi and ans>A){
                ok=true;
                break;
            }
        }while(next_permutation(begin(ans),end(ans)));

        if(!ok) {
            cout << -1 << endl;
            return;
        }
    }

    rep(i,0,N){
        cout << ans[i]+1;
        if(i==N-1) cout << endl;
        else cout << " ";
    }
}

int main() {
    std::cin.tie(0);
    std::ios::sync_with_stdio(false);
    cout.setf(ios::fixed);
    cout.precision(16);
    solve();
    return 0;
}
0