結果

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

ソースコード

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()+1,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