結果
問題 | No.2740 Old Maid |
ユーザー |
![]() |
提出日時 | 2024-04-20 13:10:48 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 453 ms / 2,000 ms |
コード長 | 2,003 bytes |
コンパイル時間 | 3,868 ms |
コンパイル使用メモリ | 227,832 KB |
最終ジャッジ日時 | 2025-02-21 06:25:39 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 62 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ll = long long;constexpr ll mod = 998244353;template<ll mod>struct Mint {using M=Mint; ll v;M& put(ll x) { v=(x<mod)?x:x-mod; return *this; }Mint(ll x=0) { put(x%mod+mod); }M operator+(M m) {return M().put(v+m.v);}M operator-(M m) {return M().put(v+mod-m.v);}M operator*(M m) {return M().put(v*m.v%mod);}M operator/(M m) {return M().put(v*m.inv().v%mod);}// BEGIN IGNOREM operator+=(M m) { return put(v+m.v); }M operator-=(M m) { return put(v+mod-m.v); }M operator*=(M m) { return put(v*m.v%mod); }M operator/=(M m) { return put(v*m.inv().v%mod); }// END IGNOREbool operator==(M m) { return v==m.v; }M pow(ll m) const {M x=v, res=1;while (m) {if (m&1) res=res*x;x=x*x; m>>=1;}return res;}M inv() { return pow(mod-2); }};using mint = Mint<mod>;#include<ext/pb_ds/assoc_container.hpp>#include<ext/pb_ds/tree_policy.hpp> // for tree#include<ext/pb_ds/tag_and_trait.hpp> // for treeusing namespace __gnu_pbds;template <typename T>using gtree = tree<T, null_type, less<T>, rb_tree_tag,tree_order_statistics_node_update>;int main(){cin.tie(0);cin.sync_with_stdio(0);int n;cin >> n;vector<int> p(n);for(int i = 0; i < n; ++i){cin >> p[i];--p[i];}vector<int> inv(n);for(int i = 0; i < n; ++i){inv[p[i]] = i;}gtree<int> gt;for(int i = 0; i < n; ++i){gt.insert(i);}int k = 0;vector<int> q;for(int iter = 0; iter < n / 2; ++iter){while(true){int key = inv[k];if(gt.find(key) != gt.end() && gt.order_of_key(key) != gt.size() - 1){int idx1 = key;int idx2 = *gt.find_by_order(gt.order_of_key(key) + 1);gt.erase(idx1);gt.erase(idx2);q.emplace_back(p[idx1]);q.emplace_back(p[idx2]);break;}++k;}}for(int i = 0; i < q.size(); ++i){cout << q[i] + 1 << " \n"[i + 1 == q.size()];}}