結果
問題 |
No.3167 [Cherry 7th Tune C] Cut in Queue
|
ユーザー |
![]() |
提出日時 | 2025-05-30 21:59:28 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 495 ms / 2,500 ms |
コード長 | 6,842 bytes |
コンパイル時間 | 2,332 ms |
コンパイル使用メモリ | 206,052 KB |
実行使用メモリ | 18,760 KB |
最終ジャッジ日時 | 2025-05-30 21:59:44 |
合計ジャッジ時間 | 16,146 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 43 |
ソースコード
//#pragma GCC target("avx2") //#pragma GCC optimize("O3") //#pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll = long long; using pii = pair<int,int>; using pll = pair<ll,ll>; using pli = pair<ll,int>; #define AMARI 998244353 //#define AMARI 1000000007 #define el '\n' #define El '\n' #define YESNO(x) ((x) ? "Yes" : "No") #define YES YESNO(true) #define NO YESNO(false) #define REV_PRIORITY_QUEUE(tp) priority_queue<tp,vector<tp>,greater<tp>> #define EXIT_ANS(x) {cout << (x) << '\n'; return;} template <typename T> void inline SORT(vector<T> &v){sort(v.begin(),v.end()); return;} template <typename T> void inline VEC_UNIQ(vector<T> &v){sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); return;} template <typename T> T inline MAX(vector<T> &v){return *max_element(v.begin(),v.end());} template <typename T> T inline MIN(vector<T> &v){return *min_element(v.begin(),v.end());} template <typename T> T inline SUM(vector<T> &v){T ans = 0; for(int i = 0; i < (int)v.size(); i++)ans += v[i]; return ans;} template <typename T> void inline DEC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]--; return;} template <typename T> void inline INC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]++; return;} void inline TEST(void){cerr << "TEST" << endl; return;} template <typename T> bool inline chmin(T &x,T y){ if(x > y){ x = y; return true; } return false; } template <typename T> bool inline chmax(T &x,T y){ if(x < y){ x = y; return true; } return false; } template <typename T = long long> vector<T> inline get_vec(int n){ vector<T> ans(n); for(int i = 0; i < n; i++)cin >> ans[i]; return ans; } template <typename T> void inline print_vec(vector<T> &vec,bool kaigyou = false){ int n = (int)vec.size(); for(int i = 0; i < n; i++){ cout << vec[i]; if(kaigyou || i == n - 1)cout << '\n'; else cout << ' '; } if(!n)cout << '\n'; return; } template <typename T> void inline debug_vec(vector<T> &vec,bool kaigyou = false){ int n = (int)vec.size(); for(int i = 0; i < n; i++){ cerr << vec[i]; if(kaigyou || i == n - 1)cerr << '\n'; else cerr << ' '; } if(!n)cerr << '\n'; return; } vector<vector<int>> inline get_graph(int n,int m = -1,bool direct = false){ if(m == -1)m = n - 1; vector<vector<int>> g(n); while(m--){ int u,v; cin >> u >> v; u--; v--; g[u].push_back(v); if(!direct)g[v].push_back(u); } return g; } vector<int> make_inv(vector<int> p){ int n = (int)p.size(); vector<int> ans(n); for(int i = 0; i < n; i++)ans[p[i]] = i; return ans; } //フェネック木(BIT とも呼ばれるアレ) operator+, operator+=,operator- さえ定義できていればオンラインに区間和を取得できる。 //オーダーはセグ木と一緒だが、定数倍が圧倒的に良い。ほぼ定数。 template <typename T> class ococo_fenwicktree{ private: int n; vector<T> a; vector<int> sz; public: //初め、全て0で埋まっている ococo_fenwicktree(int N){ n = N; a.resize(n); sz.resize(n); for(int i = 0; i < n; i++){ a[i] = 0; sz[i] = 1; int temp = i + 1; while(!(temp % 2)){ sz[i] *= 2; temp /= 2; } } //for(int i = 0; i < n; i++)cerr << sz[i] << el; } void add_num(int i,T x){ int idx = i; while(idx < n){ a[idx] += x; idx += sz[idx]; } } //[l,r] の区間和を取得する。 T get_sum(int l,int r){ if(l != 0){ return get_sum(0,r) - get_sum(0,l - 1); } T ans = 0; int idx = r; while(idx >= 0){ ans += a[idx]; idx -= sz[idx]; } return ans; } void update_num(int i,T x){ T val = get_sum(i,i); T dif = x - val; add_num(i,dif); return; } }; #define MULTI_TEST_CASE false void solve(void){ //問題を見たらまず「この問題設定から言えること」をいっぱい言う //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く //複数の解法のアイデアを思いついた時は全部メモしておく //g++ -D_GLIBCXX_DEBUG -Wall -O2 e.cpp -o o int n; cin >> n; vector<int> a = get_vec<int>(n); DEC(a); int q; cin >> q; vector<pii> query(q); for(int i = 0; i < q; i++){ int t; cin >> t; if(t == 2){ query[i] = pair(t,-1); continue; } query[i].first = t; cin >> query[i].second; } vector<int> mae(n + q,-1); for(int i = 0; i < n - 1; i++){ mae[a[i + 1]] = a[i]; } int matubi = a.back(); int b = n; for(int i = 0; i < q; i++){ if(query[i].first == 1){ int a = query[i].second; a--; if(a == -1){ mae[b] = matubi; matubi = b; b++; continue; } //b を a の直前に入れる mae[b] = mae[a]; mae[a] = b; b++; } } int idx = matubi; vector<int> vec; while(idx != -1){ vec.push_back(idx); idx = mae[idx]; } reverse(vec.begin(),vec.end()); /* INC(vec); debug_vec(vec); DEC(vec); */ vector<int> gv = make_inv(vec); int m = (int)vec.size(); ococo_fenwicktree<int> BIT(n + q); for(int i = 0; i < m; i++){ if(vec[i] < n)BIT.update_num(i,1); } //for(int i = 0; i < m; i++)cerr << BIT.get_sum(i,i) << ' '; cerr << El; int l = 0; b = n; for(int i = 0; i < q; i++){ if(query[i].first == 1){ BIT.update_num(gv[b],1); b++; continue; } if(query[i].first == 2){ BIT.update_num(l,0); l++; continue; } //for(int j = 0; j < m; j++)cerr << BIT.get_sum(j,j) << ' '; cerr << El; //[0,x] の合計が k 以上になる最小の x を求める int k = query[i].second; int ok = m - 1,ng = -1; while(ok - ng >= 2){ int mid = (ok + ng) / 2; if(BIT.get_sum(0,mid) >= k)ok = mid; else ng = mid; } cout << vec[ok] + 1 << el; } return; } void calc(void){ return; } signed main(void){ cin.tie(nullptr); ios::sync_with_stdio(false); calc(); int t = 1; if(MULTI_TEST_CASE)cin >> t; while(t--){ solve(); } return 0; }