結果
問題 | No.1051 PQ Permutation |
ユーザー |
![]() |
提出日時 | 2020-05-08 22:52:38 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 149 ms / 2,000 ms |
コード長 | 3,277 bytes |
コンパイル時間 | 1,520 ms |
コンパイル使用メモリ | 181,724 KB |
実行使用メモリ | 17,280 KB |
最終ジャッジ日時 | 2024-07-05 19:54:03 |
合計ジャッジ時間 | 5,697 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define rep(i,m,n) for(int (i)=(int)(m);i<(int)(n);i++) #define rep2(i,m,n) for(int (i)=(int)(n)-1;i>=(int)(m);i--) #define REP(i,n) rep(i,0,n) #define REP2(i,n) rep2(i,0,n) #define FOR(i,c) for(decltype((c).begin())i=(c).begin();i!=(c).end();++i) #define all(hoge) (hoge).begin(),(hoge).end() #define en '\n' using ll = long long; using ull = unsigned long long; template <class T> using vec = vector<T>; template <class T> using vvec = vector<vec<T>>; typedef pair<ll, ll> P; constexpr long long INF = 1LL << 60; constexpr int INF_INT = 1 << 25; constexpr long long MOD = (ll) 1e9 + 7; //constexpr long long MOD = 998244353LL; typedef vector<ll> Array; typedef vector<Array> Matrix; template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; } template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; } struct Edge { ll to, cap, rev; Edge(ll _to, ll _cap, ll _rev) { to = _to; cap = _cap; rev = _rev; } }; using Edges = vector<Edge>; using Graph = vector<Edges>; void add_edge(Graph& G, ll from, ll to, ll cap, bool revFlag, ll revCap) { G[from].push_back(Edge(to, cap, (ll)G[to].size())); if (revFlag)G[to].push_back(Edge(from, revCap, (ll)G[from].size() - 1)); } void solve(){ ll n,p,q; cin>>n>>p>>q; Array a(n); REP(i,n) cin>>a[i]; Array b=a; REP(k,1){ bool flag=next_permutation(all(b)); if(!flag) { cout<<-1<<en; return; } set<ll> st; REP(i,n){ if(i+1!=q) st.insert(i+1); } Array ans(n,0); bool big=false; ll hoken=-1; REP(i,n){ if(big){ auto it = st.begin(); ans[i]=*it; if(*it==p) st.insert(q);//pが出たら使える st.erase(it); }else{ auto it = st.lower_bound(b[i]); if(it==st.end()){ if(hoken==-1){ cout<<-1<<en; return; } //大きくできるところまで戻る rep(j,hoken,i){ st.insert(ans[j]); if(ans[j]==p) st.erase(q);//pを戻した場合はqは使えなくなる } i=hoken; auto it2=st.upper_bound(b[i]); ans[i]=*it2; if(*it2==p) st.insert(q); st.erase(it2); big=true; continue; } if(*it>b[i]) big=true; ans[i]=*it; if(*it==p) st.insert(q);//pが出たら使える auto it2=it; it2++; if(it2!=st.end()) hoken=i; st.erase(it); } } if(flag){ for(auto i:ans) cout<<i<<" "; cout<<en; return; } } cout<<-1<<en; } int main() { ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); solve(); //ll t;cin>>t;REP(i,t) solve(); return 0; }