結果

問題 No.1051 PQ Permutation
ユーザー mugen_1337mugen_1337
提出日時 2020-05-09 12:48:36
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 160 ms / 2,000 ms
コード長 3,316 bytes
コンパイル時間 2,260 ms
コンパイル使用メモリ 215,316 KB
実行使用メモリ 13,440 KB
最終ジャッジ日時 2024-07-05 20:12:20
合計ジャッジ時間 5,243 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 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 AC 1 ms
5,376 KB
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 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 92 ms
13,184 KB
testcase_16 AC 85 ms
12,544 KB
testcase_17 AC 34 ms
5,376 KB
testcase_18 AC 33 ms
5,376 KB
testcase_19 AC 31 ms
5,376 KB
testcase_20 AC 65 ms
9,600 KB
testcase_21 AC 68 ms
9,984 KB
testcase_22 AC 33 ms
5,376 KB
testcase_23 AC 67 ms
9,600 KB
testcase_24 AC 64 ms
9,472 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 6 ms
5,376 KB
testcase_31 AC 3 ms
5,376 KB
testcase_32 AC 6 ms
5,376 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 38 ms
8,320 KB
testcase_36 AC 37 ms
8,192 KB
testcase_37 AC 18 ms
5,376 KB
testcase_38 AC 80 ms
13,312 KB
testcase_39 AC 66 ms
13,440 KB
testcase_40 AC 30 ms
5,376 KB
testcase_41 AC 55 ms
8,576 KB
testcase_42 AC 1 ms
5,376 KB
testcase_43 AC 1 ms
5,376 KB
testcase_44 AC 1 ms
5,376 KB
testcase_45 AC 1 ms
5,376 KB
testcase_46 AC 1 ms
5,376 KB
testcase_47 AC 2 ms
5,376 KB
testcase_48 AC 160 ms
13,440 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:29:12: warning: 'qi' may be used uninitialized [-Wmaybe-uninitialized]
   29 |     int pi,qi;
      |            ^~
main.cpp:34:5: warning: 'pi' may be used uninitialized [-Wmaybe-uninitialized]
   34 |     if(pi<qi){
      |     ^~
main.cpp:29:9: note: 'pi' was declared here
   29 |     int pi,qi;
      |         ^~

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define ALL(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<(n);i++)
#define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl;
#define INF 1000000000
#define mod 1000000007
using ll=long long;
const ll LINF=1001002003004005006ll;
int dx[]={1,0,-1,0};
int dy[]={0,1,0,-1};
// ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}

signed main(){
    cin.tie(0);
    ios::sync_with_stdio(0);
    
    int n,p,q;cin>>n>>p>>q;p--,q--;
    vector<int> a(n);
    rep(i,n){
        cin>>a[i];a[i]--;
    }
    if(!next_permutation(ALL(a))){
        cout<<-1<<endl;
        return 0;
    }
    int pi,qi;
    rep(i,n){
        if(a[i]==p) pi=i;
        if(a[i]==q) qi=i;
    }
    if(pi<qi){
        rep(i,n) cout<<a[i]+1<<(i+1==n?"\n":" ");
        return 0;
    }

    sort(a.begin()+qi+1,a.end(),greater<>());

    /*
    今aは，... q ... p ...でqより後ろは降順
    ここからネクパしたときに，
    qより右にqより大きい数があればqの位置で繰り上がりが起こる． ... (A)
    qより右にqより大きい数が無ければヤバイ．これだけ別枠で考える... (B)

    (A)の場合はネクパして，もうokならいい
    ダメならネクパした後，右側は昇順になってるからpをqの左隣に持ってくれば終わり
    
    (B)の場合．
    例えば，
    p=7,q=1
    4 5 6 8 9 7 3 2 1
    からネクパ->
    4 5 6 9 8 7 3 2 1となる．p,qの位置が変わらないことあるし色々ヤベェ
    ネクパにより，
    qiから左のa[i]<a[i+1],(i+1<=q)のところでa[i]とa[i]より大きくa[i]より右側にある数
    とでswapが起こる
    それより右が昇順になる

    o o o o o o o Q o o o o P o o o o
    [ なんか順列 ][        降順      ]
    
    o o o o o<o>o>Q o o o o P o o o o
            ↑ｺｺ
              とそこより右の中でそれより大きい中の最小値とswap，
                そこから右が全部降順になっちゃっているため，
    これを繰り返し，Qより大きな値がQより右にくるまで順列を進めることを加速させることができる．
    */

    set<int> st;
    for(int i=qi+1;i<n;i++) st.insert(a[i]);
    while(true){
        int pos=qi;
        while(pos>=0 and a[pos]>a[pos+1]){
            st.insert(a[pos]);pos--;
        }
        if(pos<0){
            cout<<-1<<endl;
            return 0;
        }

        if(*st.lower_bound(a[pos])==q){
            swap(a[pos],a[qi]);
            st.erase(q);
            st.insert(a[qi]);
            qi=pos;
        }else{
            // qより右にqより大きい数が存在する
            // -> (A)に帰着した
            break;
        }
    }

    if(!next_permutation(ALL(a))){
        cout<<-1<<endl;
        return 0;
    }

    rep(i,n){
        if(a[i]==p) pi=i;
        if(a[i]==q) qi=i;
    }
    if(pi<qi){
        rep(i,n) cout<<a[i]+1<<(i+1==n?"\n":" ");
        return 0;
    }

    // p->qに
    a.erase(a.begin()+qi);
    a.insert(a.begin()+pi,q);
    rep(i,n) cout<<a[i]+1<<(i+1==n?"\n":" ");
    return 0;
}
0