結果
問題 | No.1558 Derby Live |
ユーザー |
![]() |
提出日時 | 2021-05-20 20:24:19 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 286 ms / 2,000 ms |
コード長 | 2,378 bytes |
コンパイル時間 | 1,737 ms |
コンパイル使用メモリ | 173,344 KB |
実行使用メモリ | 10,496 KB |
最終ジャッジ日時 | 2024-06-25 08:01:56 |
合計ジャッジ時間 | 10,905 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
#include <bits/stdc++.h> using namespace std; template <typename T> struct SegTree { const T INF = numeric_limits<T>::max(); int n, m; vector<vector<T>> dat; SegTree(int n_, int m_) : dat(n_ * 4, vector<T>(m_, 0)) { int x = 1; while (n_ > x) { x *= 2; } n = x; m = m_; } void init() { for (int i = 0; i < 2 * n - 1; i++) { for (int j = 0; j < m; j++) { dat[i][j] = j; } } } void update(int i, vector<T> x) { i += n - 1; for (int j = 0; j < m; j++) dat[i][j] = x[j]; while (i > 0) { i = (i - 1) / 2; for (int j = 0; j < m; j++) dat[i][j] = dat[i * 2 + 2][dat[i * 2 + 1][j]]; } } vector<T> query(int a, int b) { return query_sub(a, b, 0, 0, n); } vector<T> query_sub(int a, int b, int k, int l, int r) { vector<T> x(m); if (r <= a || b <= l) { for (int i = 0; i < m; i++) x[i] = i; } else if (a <= l && r <= b) { for (int i = 0; i < m; i++) x[i] = dat[k][i]; } else { vector<T> vl = query_sub(a, b, k * 2 + 1, l, (l + r) / 2); vector<T> vr = query_sub(a, b, k * 2 + 2, (l + r) / 2, r); for (int i = 0; i < m; i++) x[i] = vr[vl[i]]; } return x; } }; int main() { long long N, M, Q; cin >> N >> M >> Q; SegTree<long long> Seg(M, N); Seg.init(); for (int i = 0; i < Q; i++) { int com; cin >> com; if (com == 1) { long long B; cin >> B; vector<long long> P(N); for (int j = 0; j < N; j++) cin >> P[j]; for (int j = 0; j < N; j++) P[j]--; Seg.update(B - 1, P); } if (com == 2) { long long S; cin >> S; auto x = Seg.query(0, S); vector<long long> y(N); for (int j = 0; j < N; j++) y[x[j]] = j; cout << y[0] + 1; for (int j = 1; j < N; j++) cout << " " << y[j] + 1; cout << endl; } if (com == 3) { long long a, b; cin >> a >> b; auto x = Seg.query(a - 1, b); long long count = 0; for (int j = 0; j < N; j++) count += abs(x[j] - j); cout << count << endl; } } }