#793918 (C) No.2065 Sum of Min

• 問題: No.2065 Sum of Min
• ユーザー: 👑 ygussany
• 提出日時: 2022-08-29 20:46:34
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 204 ms / 2,000 ms
• コード長: 2,392 bytes
• コンパイル時間: 1,454 ms
• コンパイル使用メモリ: 33,152 KB
• 実行使用メモリ: 14,336 KB
• 最終ジャッジ日時: 2024-04-24 11:30:20
• 合計ジャッジ時間: 7,795 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	113 ms; 14,336 KB
testcase_05	AC	91 ms; 14,336 KB
testcase_06	AC	143 ms; 14,336 KB
testcase_07	AC	62 ms; 14,336 KB
testcase_08	AC	145 ms; 14,208 KB
testcase_09	AC	153 ms; 14,336 KB
testcase_10	AC	104 ms; 14,336 KB
testcase_11	AC	102 ms; 14,336 KB
testcase_12	AC	191 ms; 14,336 KB
testcase_13	AC	184 ms; 14,208 KB
testcase_14	AC	204 ms; 14,208 KB
testcase_15	AC	186 ms; 14,208 KB
testcase_16	AC	188 ms; 14,336 KB
testcase_17	AC	187 ms; 14,208 KB
testcase_18	AC	185 ms; 14,336 KB
testcase_19	AC	191 ms; 14,336 KB
testcase_20	AC	191 ms; 14,336 KB
testcase_21	AC	200 ms; 14,336 KB

ソースコード

#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;
}