結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー walkre
提出日時 2020-05-14 19:54:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 162 ms / 2,000 ms
コード長 4,511 bytes
コンパイル時間 1,911 ms
コンパイル使用メモリ 177,940 KB
実行使用メモリ 15,744 KB
最終ジャッジ日時 2024-09-15 19:06:25
合計ジャッジ時間 7,751 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

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);
    bool pf=false,qf=false;

    i64 idx=-1;
    set<i64> st;
    for(i64 i=N-1; i>=0; --i){
        if(A[i]==P) pf=true;
        if(A[i]==Q) qf=true;

        if(A[i]==P){
            if(pi<qi){
                st.insert(P);
                st.erase(Q);
            }
            if(qi<pi) ; //nop
        }else if(A[i]==Q){
            if(pi<qi) st.insert(Q);
            if(qi<pi) st.insert(P);
        }else{
            st.insert(A[i]);
        }

        if(st.upper_bound(A[i])==end(st) or (qi<pi and !qf)) continue;
        idx=i;
        break;
    }

    if(idx==-1){
        cout << -1 << endl;
        return;
    }

    vector<i64> ans(N);
    rep(i,0,idx) ans[i]=A[i];

    i64 tmp=*st.upper_bound(A[idx]);
    ans[idx]=tmp;
    st.erase(tmp);
    if(tmp==P) st.insert(Q);

    rep(i,idx+1,N){
        i64 x=*st.begin();
        ans[i]=x;
        st.erase(x);
        if(x==P) st.insert(Q);
    }

    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