結果

問題 No.2304 Distinct Elements
ユーザー ygussanyygussany
提出日時 2023-04-29 17:50:33
言語 C
(gcc 12.3.0)
結果
TLE  
実行時間 -
コード長 2,140 bytes
コンパイル時間 412 ms
コンパイル使用メモリ 32,384 KB
実行使用メモリ 13,772 KB
最終ジャッジ日時 2024-11-18 12:13:58
合計ジャッジ時間 97,659 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
10,404 KB
testcase_01 AC 1 ms
10,404 KB
testcase_02 AC 1 ms
10,532 KB
testcase_03 AC 1 ms
12,096 KB
testcase_04 TLE -
testcase_05 TLE -
testcase_06 TLE -
testcase_07 TLE -
testcase_08 TLE -
testcase_09 AC 72 ms
12,028 KB
testcase_10 AC 72 ms
9,892 KB
testcase_11 AC 73 ms
13,764 KB
testcase_12 AC 71 ms
9,888 KB
testcase_13 AC 72 ms
13,764 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 1 ms
10,276 KB
testcase_20 WA -
testcase_21 AC 1 ms
13,764 KB
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 TLE -
testcase_30 WA -
testcase_31 TLE -
testcase_32 AC 55 ms
11,612 KB
testcase_33 AC 6 ms
13,764 KB
testcase_34 AC 58 ms
10,404 KB
testcase_35 AC 66 ms
10,272 KB
testcase_36 AC 5 ms
13,760 KB
testcase_37 WA -
testcase_38 TLE -
testcase_39 TLE -
testcase_40 TLE -
testcase_41 WA -
testcase_42 TLE -
testcase_43 TLE -
testcase_44 AC 53 ms
6,816 KB
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 TLE -
testcase_50 TLE -
testcase_51 TLE -
testcase_52 TLE -
testcase_53 WA -
testcase_54 TLE -
testcase_55 TLE -
testcase_56 TLE -
testcase_57 TLE -
testcase_58 TLE -
testcase_59 TLE -
testcase_60 AC 1 ms
6,820 KB
testcase_61 AC 1 ms
11,456 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <stdlib.h>

void merge_sort(int n, int x[])
{
	static int y[200001] = {};
	if (n <= 1) return;
	merge_sort(n / 2, &(x[0]));
	merge_sort((n + 1) / 2, &(x[n/2]));
	
	int i, p, q;
	for (i = 0, p = 0, q = n / 2; i < n; i++) {
		if (p >= n / 2) y[i] = x[q++];
		else if (q >= n) y[i] = x[p++];
		else y[i] = (x[p] < x[q])? x[p++]: x[q++];
	}
	for (i = 0; i < n; i++) x[i] = y[i];
}

typedef struct {
	int key, id;
} data;

typedef struct {
	data obj[400001];
	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, A[200001];
	scanf("%d", &N);
	for (i = 0; i < N; i++) scanf("%d", &(A[i]));
	merge_sort(N, A);
	
	int j, k, mate[200001];
	max_heap h;
	data d;
	long long ans = 0;
	h.size = 0;
	mate[0] = 0;
	d.key = A[0];
	d.id = A[0];
	push(&h, d);
	for (i = 1; i < N; i++) {
		d = pop(&h);
		if (A[i] >= d.key + 1) {
			push(&h, d);
			mate[i] = 0;
			d.key = A[i];
			d.id = A[i];
			if (h.size > 0 && h.obj[1].key == d.id - 1) {
				h.obj[1].key = d.key;
			} else push(&h, d);
		} else {
			for (j = i - (d.key - d.id + 1), k = 0; j < i; j++) if (mate[j] > 0) k++;
			if (abs(d.id - 1 - A[i]) - k < abs(d.key + 1 - A[i])) {
				for (j = i - (d.key - d.id + 1), k = 0; j < i; j++) mate[j]--;
				mate[i] = d.key - A[i];
				d.id--;
				if (h.size > 0 && h.obj[1].key == d.id - 1) {
					h.obj[1].key = d.key;
				} else push(&h, d);
			} else {
				mate[i] = ++d.key - A[i];
				push(&h, d);
			}
		}
	}
	for (i = 0; i < N; i++) ans += abs(mate[i]);
	printf("%lld\n", ans);
	fflush(stdout);
	return 0;
}
0