結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー xenon_motsu
提出日時 2020-05-08 23:53:27
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,894 bytes
コンパイル時間 2,582 ms
コンパイル使用メモリ 198,868 KB
最終ジャッジ日時 2025-01-10 09:25:50
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
template<class T,class U> inline bool chmin(T&x,U y){if(x>y){x=y;return true;}return false;}
template<class T,class U> inline bool chmax(T&x,U y){if(x<y){x=y;return true;}return false;}
#define fr(i,n) for(int i=0;i<(n);++i)
#define Fr(i,n) for(int i=1;i<=(n);++i)
#define ifr(i,n) for(int i=(n)-1;i>=0;--i)
#define iFr(i,n) for(int i=(n);i>0;--i)

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n,p,q;
    cin>>n>>p>>q;
    vector<int> a(n+1);
    Fr(i,n) cin>>a[i];
    if(!next_permutation(a.begin(),a.end())){
        cout<<-1<<endl;
        return 0;
    }
    int t{1};
    Fr(i,n){
        if(i&&a[i-1]<a[i]) t=i;
        if(a[i]==p){
            Fr(j,n) cout<<a[j]<<" ";
            cout<<endl;
            return 0;
        }
        if(a[i]==q){
            sort(a.begin()+i+1,a.end());
            reverse(a.begin()+i+1,a.end());
            if(q<a[i+1]) t=i+1;

            if(t==1){
                cout<<-1<<endl;
                return 0;
            }
            reverse(a.begin()+t,a.end());
            int I=upper_bound(a.begin()+t,a.end(),a[t-1])-a.begin();
            swap(a[I],a[t-1]);
            if(a[t-1]==p){
                Fr(j,n) cout<<a[j]<<" ";
                cout<<endl;
                return 0;
            }
            for(int s=t;s<=n;++s){
                if(a[s]==p){
                    Fr(j,n) cout<<a[j]<<" ";
                    cout<<endl;
                    return 0;
                }
                if(a[s]==q){
                    Fr(j,s-1) cout<<a[j]<<" ";
                    int j=s+1;
                    for(;j<=n&&a[j]<=p;++j) cout<<a[j]<<" ";
                    cout<<q<<" ";
                    for(;j<=n;++j) cout<<a[j]<<" ";
                    cout<<endl;
                    return 0;
                }
            }

        }
    }
}
0