結果

問題 No.877 Range ReLU Query
ユーザー snrnsidy
提出日時 2021-05-09 03:24:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 100 ms / 2,000 ms
コード長 1,485 bytes
コンパイル時間 2,411 ms
コンパイル使用メモリ 208,264 KB
最終ジャッジ日時 2025-01-21 09:33:14
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

long long int tree[2][100005];
long long int sum(int i,int idx)
{
	long long int ans = 0;
	while (i > 0)
	{
		ans += tree[idx][i];
		i -= (i & -i);
	}
	return ans;
}

void update(int i, long long int diff,int idx)
{
	while (i < 100005)
	{
		tree[idx][i] += diff;
		i += (i & (-i));
	}
}

long long int res[100005];

int main(void)
{
	cin.tie(0);
	ios::sync_with_stdio(false);

	priority_queue <pair<long long int, int>> pque;
	vector <pair<pair<int,int>, pair<int, int>>> query;
	int n, q, a;
	
	cin >> n >> q;

	for (int i = 1; i <= n; i++)
	{
		cin >> a;
		pque.push(make_pair(a, i));
	}

	for (int i = 0; i < q; i++)
	{
		int x,l, r;
		cin >> x >> l >> r >> a;
		query.push_back(make_pair(make_pair(a,i), make_pair(l, r)));
	}

	sort(query.rbegin(), query.rend());

	for (int i = 0; i < q; i++)
	{
		while (!pque.empty())
		{
			if (pque.top().first < query[i].first.first)
			{
				break;
			}
			//cout << pque.top().first << ' ' << pque.top().second << ' ' << query[i].first.first << '\n';
			update(pque.top().second, pque.top().first, 0);
			update(pque.top().second, 1, 1);
			pque.pop();
		}
		int l = query[i].second.first;
		int r = query[i].second.second;
		int idx = query[i].first.second;
		long long int val = sum(r, 0) - sum(l - 1, 0);
		long long int num = sum(r, 1) - sum(l - 1, 1);
		val -= (num * query[i].first.first);
		res[idx] = val;
	}

	for (int i = 0; i < q; i++)
	{
		cout << res[i] << '\n';
	}

	return 0;
}
0