結果

問題 No.2623 Room Allocation
ユーザー 👑 ygussany
提出日時 2024-02-09 21:37:46
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 56 ms / 2,000 ms
コード長 1,373 bytes
コンパイル時間 237 ms
コンパイル使用メモリ 30,720 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-28 14:51:18
合計ジャッジ時間 2,157 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>

typedef struct {
	long long key;
	int id;
} data;

typedef struct {
	data obj[200001];
	int size;
} max_heap;

void push(max_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(max_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, X, Y, P[200001];
	char c[200001];
	scanf("%d %d %d", &N, &X, &Y);
	for (i = 1; i <= N; i++) scanf("%d %c", &(P[i]), &(c[i]));
	
	int j;
	long long ans = 0, tmp[2];
	max_heap h;
	data d;
	h.size = 0;
	for (i = 1; i <= N && i <= X + Y; i++) {
		for (j = i, tmp[0] = 0, tmp[1] = 0; j <= N; j += X + Y) {
			if (c[j] == 'A') tmp[0] += P[j];
			else tmp[1] += P[j];
		}
		d.key = tmp[0] - tmp[1];
		push(&h, d);
		ans += tmp[1];
	}
	while (X > 0 && h.size > 0) {
		if (h.obj[1].key <= 0 && h.size <= Y) break;
		X--;
		d = pop(&h);
		ans += d.key;
	}
	printf("%lld\n", ans);
	fflush(stdout);
	return 0;
}
0