結果

問題 No.2065 Sum of Min
ユーザー ygussanyygussany
提出日時 2022-08-29 20:46:34
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 241 ms / 2,000 ms
コード長 2,392 bytes
コンパイル時間 1,220 ms
コンパイル使用メモリ 31,488 KB
実行使用メモリ 14,368 KB
最終ジャッジ日時 2024-11-06 19:15:43
合計ジャッジ時間 8,060 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
8,060 KB
testcase_01 AC 2 ms
8,056 KB
testcase_02 AC 2 ms
8,068 KB
testcase_03 AC 2 ms
7,940 KB
testcase_04 AC 123 ms
14,292 KB
testcase_05 AC 96 ms
14,300 KB
testcase_06 AC 159 ms
14,356 KB
testcase_07 AC 62 ms
14,292 KB
testcase_08 AC 168 ms
14,328 KB
testcase_09 AC 185 ms
14,340 KB
testcase_10 AC 114 ms
14,320 KB
testcase_11 AC 123 ms
14,360 KB
testcase_12 AC 225 ms
14,228 KB
testcase_13 AC 228 ms
14,296 KB
testcase_14 AC 215 ms
14,316 KB
testcase_15 AC 241 ms
14,288 KB
testcase_16 AC 215 ms
14,360 KB
testcase_17 AC 229 ms
14,368 KB
testcase_18 AC 225 ms
14,356 KB
testcase_19 AC 219 ms
14,332 KB
testcase_20 AC 224 ms
14,344 KB
testcase_21 AC 228 ms
14,296 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

int leaf[100001];

typedef struct {
	int left, right;
	long long sum;
} seg_node;

void init_node(seg_node v[], int k, int l, int r)
{
	v[k].left = l;
	v[k].right = r;
	v[k].sum = 0;
	if (l < r) {
		init_node(v, k << 1, l, (l + r) / 2);
		init_node(v, (k << 1) ^ 1, (l + r) / 2 + 1, r);
	} else leaf[l] = k;
}

void update_node(seg_node v[], int k, int x)
{
	int i, j = leaf[k];
	v[j].sum = x;
	for (i = j >> 1; i > 0; j = i, i >>= 1) v[i].sum = v[j].sum + v[j^1].sum;
}
 
long long get_sum(seg_node v[], int k, int l, int r)
{
	if (v[k].right < l || v[k].left > r) return 0;
	else if (v[k].left >= l && v[k].right <= r) return v[k].sum;
	else return get_sum(v, k << 1, l, r) + get_sum(v, (k << 1) ^ 1, l, r);
}

typedef struct {
	int key, id;
} data;

typedef struct {
	data obj[100001];
	int size;
} min_heap;

void push(min_heap* h, data x)
{
	int i = ++(h->size), j = i >> 1;
	data tmp;
	h->obj[i] = x;
	while (j > 0) {
		if (h->obj[i].key < h->obj[j].key) {
			tmp = h->obj[j];
			h->obj[j] = h->obj[i];
			h->obj[i] = tmp;
			i = j;
			j >>= 1;
		} else break;
	}
}

data pop(min_heap* h)
{
	int i = 1, j = 2;
	data output = h->obj[1], tmp;
	h->obj[1] = h->obj[(h->size)--];
	while (j <= h->size) {
		if (j < h->size && h->obj[j^1].key < h->obj[j].key) j ^= 1;
		if (h->obj[j].key < h->obj[i].key) {
			tmp = h->obj[j];
			h->obj[j] = h->obj[i];
			h->obj[i] = tmp;
			i = j;
			j <<= 1;
		} else break;
	}
	return output;
}

int main()
{
	int i, N, Q, A[100001], L[100001], R[100001], X[100001];
	scanf("%d %d", &N, &Q);
	for (i = 1; i <= N; i++) scanf("%d", &(A[i]));
	for (i = 1; i <= Q; i++) scanf("%d %d %d", &(L[i]), &(R[i]), &(X[i]));
	
	min_heap hA, hQ;
	data d, dd;
	hA.size = 0;
	hQ.size = 0;
	for (i = 1; i <= N; i++) {
		d.key = A[i];
		d.id = i;
		push(&hA, d);
	}
	for (i = 1; i <= Q; i++) {
		d.key = X[i];
		d.id = i;
		push(&hQ, d);
	}
	
	long long ans[100001];
	seg_node v[2][262144];
	init_node(v[0], 1, 1, N);
	init_node(v[1], 1, 1, N);
	for (i = 1; i <= N; i++) update_node(v[1], i, 1);
	while (hQ.size > 0) {
		d = pop(&hQ);
		while (hA.size > 0 && hA.obj[1].key < d.key) {
			dd = pop(&hA);
			i = dd.id;
			update_node(v[0], i, A[i]);
			update_node(v[1], i, 0);
		}
		i = d.id;
		ans[i] = get_sum(v[0], 1, L[i], R[i]) + get_sum(v[1], 1, L[i], R[i]) * X[i];
	}
	for (i = 1; i <= Q; i++) printf("%lld\n", ans[i]);
	fflush(stdout);
	return 0;
}
0