結果

問題 No.1051 PQ Permutation
ユーザー mugen_1337
提出日時 2020-05-09 12:48:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 208 ms / 2,000 ms
コード長 3,316 bytes
コンパイル時間 2,399 ms
コンパイル使用メモリ 207,120 KB
最終ジャッジ日時 2025-01-10 09:45:59
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
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;
      |         ^~
main.cpp:34:5: warning: ‘qi’ may be used uninitialized [-Wmaybe-uninitialized]
   34 |     if(pi<qi){
      |     ^~
main.cpp:29:12: note: ‘qi’ 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