結果

問題 No.877 Range ReLU Query
ユーザー MarcusAureliusAntoninusMarcusAureliusAntoninus
提出日時 2019-09-06 22:41:31
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 1,950 bytes
コンパイル時間 2,164 ms
コンパイル使用メモリ 203,788 KB
実行使用メモリ 13,756 KB
最終ジャッジ日時 2024-04-25 22:04:25
合計ジャッジ時間 4,553 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 96 ms
12,272 KB
testcase_12 AC 77 ms
11,280 KB
testcase_13 AC 64 ms
10,116 KB
testcase_14 AC 66 ms
10,028 KB
testcase_15 AC 91 ms
13,252 KB
testcase_16 AC 86 ms
12,808 KB
testcase_17 AC 86 ms
13,056 KB
testcase_18 AC 88 ms
13,120 KB
testcase_19 AC 87 ms
13,756 KB
testcase_20 AC 89 ms
13,556 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

///////////////////////////
// Range Sum Query (BIT) //
///////////////////////////

class RSumQBIT {
private:
	std::vector<int64_t> container_;

	int64_t getHelper(const int index) const
	{
		if (index < 0) return 0;
		if ((int)(container_.size()) <= index) return container_.back();
		int64_t sum{};
		for (int add_place{index}; add_place > 0; add_place -= add_place & -add_place)
			sum += container_[add_place];
		return sum;
	}

public:
	RSumQBIT(const int array_size)
		: container_(array_size + 1) {}
	// indexは0-indexed
	void update(const int index, const int64_t added)
	{
		for (int update_place{index + 1}; update_place < (int)(container_.size()); update_place += update_place & -update_place)
			container_[update_place] += added;
	}
	// left,rightは0-indexed、[left, right)の半開区間
	int64_t get(const int left, const int right) const
	{
		return -getHelper(left) + getHelper(right);
	}
};

int main()
{
	int N, Q;
	scanf("%d%d", &N, &Q);
	std::vector<std::array<int64_t, 5>> queries(N + Q);
	// {1, l, r, x}
	// {2, index, a}
	for (int64_t i{}; i < N; i++)
	{
		int64_t a;
		scanf("%lld", &a);
		queries[i] = {2ll, i, i, a, i};
	}
	for (int i{}; i < Q; i++)
	{
		int64_t com, l, r, x;
		scanf("%lld%lld%lld%lld", &com, &l, &r, &x);
		l--;
		queries[N + i] = {com, l, r, x, i};
	}

	std::sort(queries.begin(), queries.end(), [](auto& a, auto& b) { return a[3] > b[3]; });

	RSumQBIT sum(N), exist(N);
	std::vector<int64_t> ans(Q);
	for (auto& query: queries)
	{
		// for (int i{}; i < N; i++)
		// 	printf("%lld ", sum.get(i, i + 1));
		// putchar('\n');
		// for (int i{}; i < N; i++)
		// 	printf("%lld ", exist.get(i, i + 1));
		// putchar('\n');
		if (query[0] == 1)
			ans[query[4]] = sum.get(query[1], query[2]) - exist.get(query[1], query[2]) * query[3];
		else
		{
			sum.update(query[1], query[3]);
			exist.update(query[1], 1);
		}
	}
	for (auto& e: ans)
		printf("%lld\n", e);

	return 0;
}
0