結果
問題 |
No.833 かっこいい電車
|
ユーザー |
![]() |
提出日時 | 2019-05-24 23:31:09 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 187 ms / 2,000 ms |
コード長 | 1,858 bytes |
コンパイル時間 | 1,029 ms |
コンパイル使用メモリ | 87,852 KB |
実行使用メモリ | 10,368 KB |
最終ジャッジ日時 | 2024-07-02 03:47:35 |
合計ジャッジ時間 | 4,784 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include <algorithm> #include <cassert> #include <iostream> #include <set> #include <vector> #define rep(i, n) for (int i = 0; i < (int)(n); i++) using namespace std; struct SegmentTree { int n = 1; vector<int64_t> dat; SegmentTree(int N) { while (n < N) { n *= 2; } dat.resize(n * 2 - 1, 0); } void add(int i, int64_t x) { dat[i += n - 1] += x; while (i > 0) { i = (i - 1) / 2; dat[i] = dat[i * 2 + 1] + dat[i * 2 + 2]; } } int64_t sum(int l, int r) { return sum(l, r, 0, 0, n); } int64_t sum(int ql, int qr, int i, int il, int ir) { if (qr <= il or ir <= ql) return 0; if (ql <= il and ir <= qr) return dat[i]; auto m = (il + ir) / 2; return sum(ql, qr, i * 2 + 1, il, m) + sum(ql, qr, i * 2 + 2, m, ir); } }; int main() { int n, q; cin >> n >> q; vector<int> a(n); for (auto &e : a) { cin >> e; } SegmentTree s(n); rep(i, n) s.add(i, a[i]); set<pair<int, int>> seqs; rep(i, n) seqs.insert({i, i}); while (q--) { int t, x; cin >> t >> x; x--; if (t == 1) { auto itl = seqs.upper_bound({x, n}), itr = seqs.upper_bound({x + 1, n}); itl--; itr--; if (itl == itr) continue; auto nxt = make_pair(itl->first, itr->second); seqs.erase(itl); seqs.erase(itr); seqs.insert(nxt); } else if (t == 2) { auto itl = seqs.upper_bound({x, n}), itr = seqs.upper_bound({x + 1, n}); itl--; itr--; if (itl != itr) continue; auto nxtl = make_pair(itl->first, x), nxtr = make_pair(x + 1, itr->second); seqs.erase(itl); seqs.insert(nxtl); seqs.insert(nxtr); } else if (t == 3) { s.add(x, 1); } else { auto it = seqs.upper_bound({x, n}); it--; cout << s.sum(it->first, it->second + 1) << endl; } } return 0; }