#377660 (C++14) No.789 範囲の合計

• 問題: No.789 範囲の合計
• ユーザー: toma
• 提出日時: 2019-09-09 01:35:59
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 870 ms / 1,000 ms
• コード長: 1,551 bytes
• コンパイル時間: 1,793 ms
• コンパイル使用メモリ: 177,140 KB
• 実行使用メモリ: 45,944 KB
• 最終ジャッジ日時: 2024-06-27 15:23:55
• 合計ジャッジ時間: 9,527 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	772 ms; 42,472 KB
testcase_03	AC	196 ms; 5,376 KB
testcase_04	AC	870 ms; 43,000 KB
testcase_05	AC	663 ms; 42,472 KB
testcase_06	AC	713 ms; 42,468 KB
testcase_07	AC	370 ms; 5,376 KB
testcase_08	AC	766 ms; 45,944 KB
testcase_09	AC	713 ms; 43,128 KB
testcase_10	AC	541 ms; 25,700 KB
testcase_11	AC	509 ms; 42,596 KB
testcase_12	AC	521 ms; 42,476 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB

ソースコード

#include"bits/stdc++.h"
using namespace std;
#define REP(k,m,n) for(int (k)=(m);(k)<(n);(k)++)
#define rep(i,n) REP((i),0,(n))
using ll = long long;

constexpr ll n = 1ll << 30; // 横幅

//template<typename T>
class DynamicSegmentTree {
private:
	//using F = function<T(T, T)>; // モノイド型
	//const F f;   // モノイド
	//const T e;   // モノイド単位元
	unordered_map<int, ll> data;

public:
	// init忘れに注意
	DynamicSegmentTree() {}

	void add_val(int idx, ll val) {
		idx += n - 1;
		if (data.find(idx) == data.end())data[idx] = 0;
		data[idx] += val;
		while (idx > 0) {
			idx = (idx - 1) / 2;
			int num = 2 * idx + 1;
			data[idx] =
				(data.find(num) == data.end() ? 0 : data[num]) +
				(data.find(num + 1) == data.end() ? 0 : data[num + 1]);
		}
	}

	ll query(int a, int b, int k = 0, int l = 0, int r = n) {
		if (r <= a || b <= l)return 0;
		if (a <= l && r <= b) {
			return data.find(k) == data.end() ? 0 : data[k];
		}
		else {
			ll vl = data.find(2 * k + 1) == data.end()
				? 0
				: query(a, b, 2 * k + 1, l, (l + r) / 2);
			ll vr = data.find(2 * k + 2) == data.end()
				? 0
				: query(a, b, 2 * k + 2, (l + r) / 2, r);
			return vl + vr;
		}
	}
};

int main()
{
	int n;
	cin >> n;

	//function<ll(ll, ll)> f = [](ll a, ll b) {return a + b; };
	DynamicSegmentTree dst;

	ll res = 0;
	while (n--) {
		int com, x, y;
		scanf("%d %d %d", &com, &x, &y);
		//cin >> com >> x >> y;
		if (com == 0) {
			dst.add_val(x, y);
		}
		else {
			res += dst.query(x, y + 1);
		}
	}
	cout << res << endl;
	return 0;
}