結果

問題 No.789 範囲の合計
ユーザー AC2K
提出日時 2023-03-06 21:10:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 223 ms / 1,000 ms
コード長 2,908 bytes
コンパイル時間 2,344 ms
コンパイル使用メモリ 198,516 KB
最終ジャッジ日時 2025-02-11 05:56:11
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

template<class S,S(*op)(S,S),S(*e)()>
class dynamic_segtree {
public:
	dynamic_segtree(const size_t& n) :n(n), root(nullptr) {}
private:
	struct node {
		S val;
		node* left;
		node* right;

		node(const S& v) :val(v), left(nullptr), right(nullptr) {}
	};
	node* root;
	size_t n;
public:
	void update(const size_t& p, const S& x) { 
		assert(0 <= p && p < n);
		internal_update(root, 0, n, p, x); 
	}
	void add(const size_t& p, const S& x) { 
		assert(0 <= p && p < n);
		internal_add(root, 0, n, p, x); 
	}
	S operator[](const size_t& p) { 
		assert(0 <= p && p < n);
		return internal_access(root, 0, n, p); 
	}
	S prod(const size_t& l, const size_t& r) {
		assert(0 <= l && l <= r && r <= n);
		if (l == r) {
			return e();
		}

		return internal_prod(root, 0, n, l, r);
	}

private:
	void internal_update(node*& p, const size_t& l, const size_t& r, const  size_t& idx, const S& new_val) {
		if (p == nullptr) {
			p = new node(e());
		}

		if (r - l == 1) {
			p->val = new_val;
			return;
		}

		size_t mid = (l + r) >> 1;
		if (idx < mid) internal_update(p->left, l, mid, idx, new_val);
		else internal_update(p->right, mid, r, idx, new_val);
		p->val = e();
		if (p->left!=nullptr) p->val = op(p->left->val, p->val);
		if (p->right!=nullptr) p->val = op(p->val, p->right->val);
	}
	void internal_add(node*& p, const size_t& l, const size_t& r, const  size_t& idx, const S& new_val) {
		if (p == nullptr) {
			p = new node(e());
		}

		if (r - l == 1) {
			p->val = op(p->val, new_val);;
			return;
		}

		size_t mid = (l + r) >> 1;
		if (idx < mid) internal_add(p->left, l, mid, idx, new_val);
		else internal_add(p->right, mid, r, idx, new_val);
		p->val = e();
		if (p->left!=nullptr) p->val = op(p->left->val, p->val);
		if (p->right!=nullptr) p->val = op(p->val, p->right->val);
	}


	S internal_access(node*& p, const size_t& l, const size_t& r, const  size_t& idx) {
		if (p == nullptr) {
			return e();
		}
		if (r - l == 1) {
			return p->val;
		}

		size_t mid = (l + r) >> 1;
		if (idx < mid) return internal_access(p->left, l, mid, idx);
		else return internal_access(p->right, mid, r, idx);
	}

	S internal_prod(node*& p, const size_t& l, const size_t& r, const size_t& L, const size_t& R) {
		if (p == nullptr || r <= L || R <= l) {
			return e();
		}
		if (L <= l && r <= R) {
			return p->val;
		}

		size_t mid = (l + r) >> 1;
		return op(internal_prod(p->left, l, mid, L, R), internal_prod(p->right, mid, r, L, R));
	}
};


int op(int x, int y) { return x + y; }
int e() { return 0; }
int main() {
	const size_t n = 1000000001;
	dynamic_segtree<int, op, e> seg(n);

	int q;
	cin >> q;
	long long ans = 0;
	while (q--) {
		int type;
		cin >> type;
		if (type == 0) {
			size_t x;
			long long y;
			cin >> x >> y;
			seg.add(x, y);
		}
		else {
			size_t l, r;
			cin >> l >> r;
			ans += seg.prod(l, r + 1);
		}
	}
	cout << ans << '\n';
}
0