結果
問題 | No.1558 Derby Live |
ユーザー | chocorusk |
提出日時 | 2021-06-25 22:12:49 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 274 ms / 2,000 ms |
コード長 | 2,617 bytes |
コンパイル時間 | 3,575 ms |
コンパイル使用メモリ | 188,516 KB |
最終ジャッジ日時 | 2025-01-22 12:19:42 |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #include <list> #include <atcoder/all> #define popcount __builtin_popcount using namespace std; using namespace atcoder; typedef long long ll; typedef pair<int, int> P; template<typename Monoid> struct SegmentTree{ using F=function<Monoid(Monoid, Monoid)>; int sz; vector<Monoid> seg; const F f; const Monoid e; SegmentTree(int n, const F f, const Monoid &e): f(f), e(e){ sz=1; while(sz<n) sz<<=1; seg.resize(2*sz, e); } SegmentTree(int n, const F f, const Monoid &e, vector<Monoid> v): f(f), e(e){ sz=1; while(sz<n) sz<<=1; seg.resize(2*sz, e); for(int i=0; i<n; i++) seg[i+sz]=v[i]; for(int i=sz-1; i>=1; i--){ seg[i]=f(seg[2*i], seg[2*i+1]); } } void update(int k, const Monoid &x){ k+=sz; seg[k]=x; while(k>1){ k>>=1; seg[k]=f(seg[2*k], seg[2*k+1]); } } Monoid query(int a, int b){ //演算が非可換のとき a+=sz, b+=sz; Monoid retl=e, retr=e; for(;a<b; a>>=1, b>>=1){ if(b&1) retr=f(seg[--b], retr); if(a&1) retl=f(retl, seg[a++]); } return f(retl, retr); } Monoid operator[](const int &k) const{ return seg[k+sz]; } }; int main() { int n, m, q; cin>>n>>m>>q; auto f=[&](vector<int> a, vector<int> b){ vector<int> res(n); for(int i=0; i<n; i++) res[i]=b[a[i]]; return res; }; vector<int> e(n); for(int i=0; i<n; i++) e[i]=i; SegmentTree<vector<int>> seg(m, f, e); while(q--){ int t; cin>>t; if(t==1){ int d; cin>>d; vector<int> p(n); for(int i=0; i<n; i++){ cin>>p[i]; p[i]--; } seg.update(d-1, p); }else if(t==2){ int s; cin>>s; auto p=seg.query(0, s); vector<int> ans(n); for(int i=0; i<n; i++) ans[p[i]]=i; for(int i=0; i<n; i++){ cout<<ans[i]+1; if(i<n-1) cout<<" "; } cout<<endl; }else{ int l, r; cin>>l>>r; l--; auto p=seg.query(l, r); int ans=0; for(int i=0; i<n; i++) ans+=abs(i-p[i]); cout<<ans<<endl; } } return 0; }