結果

問題 No.1051 PQ Permutation
ユーザー walkrewalkre
提出日時 2020-05-09 00:00:32
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,258 bytes
コンパイル時間 1,441 ms
コンパイル使用メモリ 177,424 KB
実行使用メモリ 15,488 KB
最終ジャッジ日時 2024-07-05 20:05:56
合計ジャッジ時間 4,928 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 WA -
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 WA -
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 121 ms
15,488 KB
testcase_16 AC 120 ms
14,848 KB
testcase_17 AC 33 ms
6,400 KB
testcase_18 AC 33 ms
6,272 KB
testcase_19 AC 33 ms
6,400 KB
testcase_20 AC 81 ms
11,904 KB
testcase_21 AC 85 ms
12,160 KB
testcase_22 AC 33 ms
6,272 KB
testcase_23 AC 81 ms
11,904 KB
testcase_24 AC 79 ms
11,648 KB
testcase_25 AC 6 ms
5,376 KB
testcase_26 AC 3 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 6 ms
5,376 KB
testcase_29 AC 3 ms
5,376 KB
testcase_30 AC 8 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 8 ms
5,376 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 51 ms
9,472 KB
testcase_36 AC 53 ms
9,472 KB
testcase_37 AC 89 ms
14,208 KB
testcase_38 WA -
testcase_39 AC 85 ms
14,208 KB
testcase_40 WA -
testcase_41 AC 67 ms
10,880 KB
testcase_42 AC 1 ms
5,376 KB
testcase_43 AC 2 ms
5,376 KB
testcase_44 AC 1 ms
5,376 KB
testcase_45 AC 2 ms
5,376 KB
testcase_46 WA -
testcase_47 AC 2 ms
5,376 KB
testcase_48 AC 90 ms
14,208 KB
権限があれば一括ダウンロードができます

ソースコード

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);

    i64 idx=-1;
    set<i64> st;
    for(i64 i=N-1; i>=0; --i){
        if(A[i]==P) pi=inf;
        if(A[i]==Q) qi=inf;

        if(A[i]!=P and A[i]!=Q) st.insert(A[i]);
        else if(pi<=qi) st.insert(P);

        if(st.upper_bound(A[i])==end(st) or pi>qi) 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