#481392 (C++14) No.1051 PQ Permutation

提出ソース

結果

問題	No.1051 PQ Permutation
ユーザー	walkre
提出日時	2020-05-14 19:54:21
言語	C++14 (gcc 12.3.0 + boost 1.83.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

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	3 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	2 ms 5,376 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	162 ms 15,616 KB
testcase_16	AC	131 ms 14,720 KB
testcase_17	AC	38 ms 6,272 KB
testcase_18	AC	39 ms 6,272 KB
testcase_19	AC	39 ms 6,400 KB
testcase_20	AC	95 ms 11,904 KB
testcase_21	AC	99 ms 12,160 KB
testcase_22	AC	38 ms 6,400 KB
testcase_23	AC	97 ms 11,776 KB
testcase_24	AC	95 ms 11,776 KB
testcase_25	AC	8 ms 5,376 KB
testcase_26	AC	3 ms 5,376 KB
testcase_27	AC	2 ms 5,376 KB
testcase_28	AC	7 ms 5,376 KB
testcase_29	AC	4 ms 5,376 KB
testcase_30	AC	9 ms 5,376 KB
testcase_31	AC	3 ms 5,376 KB
testcase_32	AC	9 ms 5,376 KB
testcase_33	AC	2 ms 5,376 KB
testcase_34	AC	2 ms 5,376 KB
testcase_35	AC	60 ms 9,472 KB
testcase_36	AC	62 ms 9,344 KB
testcase_37	AC	106 ms 14,080 KB
testcase_38	AC	133 ms 15,744 KB
testcase_39	AC	100 ms 14,080 KB
testcase_40	AC	79 ms 10,880 KB
testcase_41	AC	78 ms 11,008 KB
testcase_42	AC	2 ms 5,376 KB
testcase_43	AC	2 ms 5,376 KB
testcase_44	AC	2 ms 5,376 KB
testcase_45	AC	2 ms 5,376 KB
testcase_46	AC	2 ms 5,376 KB
testcase_47	AC	2 ms 5,376 KB
testcase_48	AC	103 ms 14,080 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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;
}

yukicoder

結果

テストケース

ソースコード