結果
問題 | No.833 かっこいい電車 |
ユーザー |
![]() |
提出日時 | 2019-05-24 23:21:28 |
言語 | C++11(廃止可能性あり) (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 179 ms / 2,000 ms |
コード長 | 1,319 bytes |
コンパイル時間 | 693 ms |
コンパイル使用メモリ | 71,864 KB |
実行使用メモリ | 10,624 KB |
最終ジャッジ日時 | 2024-07-02 03:46:58 |
合計ジャッジ時間 | 4,564 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include <iostream> #include <set> #include <vector> #define llint long long using namespace std; struct SegTree{ int size; vector<llint> seg; SegTree(){} SegTree(int size){ this->size = size; seg.resize(1<<(size+1)); } void init() { for(int i = 0; i < (1<<(size+1)); i++) seg[i] = 0; } void update(int i, llint val) { i += (1 << size); seg[i] = val; while(i > 1){ i /= 2; seg[i] = (seg[i*2] + seg[i*2+1]); } } llint query(int a, int b, int k, int l, int r) { if(b < l || r < a) return 0; if(a <= l && r <= b) return seg[k]; llint lval = query(a, b, k*2, l, (l+r)/2); llint rval = query(a, b, k*2+1, (l+r)/2+1, r); return (lval + rval); } llint query(int a, int b) { return query(a, b, 1, 0, (1<<size)-1); } }; llint n, Q; llint a[100005]; set<llint> S; SegTree seg(17); int main(void) { cin >> n >> Q; for(int i = 1; i <= n; i++) cin >> a[i]; for(int i = 1; i <= n; i++) S.insert(i); S.insert(n+1); for(int i = 1; i <= n; i++) seg.update(i, a[i]); llint t, x; for(int q = 0; q < Q; q++){ cin >> t >> x; if(t == 1) S.erase(x+1); if(t == 2) S.insert(x+1); if(t == 3) seg.update(x, seg.query(x, x)+1); if(t == 4){ auto p = S.upper_bound(x); llint r = *p; p--; llint l = *p; cout << seg.query(l, r-1) << endl; } } return 0; }