結果

問題 No.738 平らな農地
ユーザー bal4ubal4u
提出日時 2019-08-04 17:08:58
言語 C
(gcc 12.3.0)
結果
TLE  
実行時間 -
コード長 2,519 bytes
コンパイル時間 1,054 ms
コンパイル使用メモリ 32,640 KB
実行使用メモリ 9,324 KB
最終ジャッジ日時 2024-07-08 10:47:47
合計ジャッジ時間 71,179 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 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 1 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 1 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 1 ms
5,376 KB
testcase_15 TLE -
testcase_16 TLE -
testcase_17 AC 1,244 ms
5,376 KB
testcase_18 AC 1,318 ms
5,376 KB
testcase_19 AC 77 ms
5,376 KB
testcase_20 TLE -
testcase_21 AC 1,163 ms
5,376 KB
testcase_22 TLE -
testcase_23 AC 894 ms
5,376 KB
testcase_24 AC 410 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 1 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 1 ms
5,376 KB
testcase_29 AC 1 ms
5,376 KB
testcase_30 AC 1 ms
5,376 KB
testcase_31 AC 1 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 1 ms
5,376 KB
testcase_34 AC 1 ms
5,376 KB
testcase_35 AC 1 ms
5,376 KB
testcase_36 AC 1 ms
5,376 KB
testcase_37 AC 1 ms
5,376 KB
testcase_38 AC 1 ms
5,376 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 1 ms
5,376 KB
testcase_41 AC 1 ms
5,376 KB
testcase_42 AC 1 ms
5,376 KB
testcase_43 AC 1 ms
5,376 KB
testcase_44 AC 2 ms
5,376 KB
testcase_45 TLE -
testcase_46 TLE -
testcase_47 TLE -
testcase_48 TLE -
testcase_49 TLE -
testcase_50 TLE -
testcase_51 TLE -
testcase_52 TLE -
testcase_53 TLE -
testcase_54 TLE -
testcase_55 TLE -
testcase_56 TLE -
testcase_57 TLE -
testcase_58 TLE -
testcase_59 TLE -
testcase_60 TLE -
testcase_61 TLE -
testcase_62 TLE -
testcase_63 TLE -
testcase_64 TLE -
testcase_65 AC 1 ms
6,940 KB
testcase_66 AC 1 ms
6,948 KB
testcase_67 TLE -
testcase_68 TLE -
testcase_69 AC 841 ms
6,944 KB
testcase_70 TLE -
testcase_71 WA -
testcase_72 WA -
testcase_73 TLE -
testcase_74 WA -
testcase_75 AC 1,062 ms
6,944 KB
testcase_76 TLE -
testcase_77 AC 217 ms
6,944 KB
testcase_78 AC 234 ms
6,944 KB
testcase_79 AC 792 ms
6,944 KB
testcase_80 TLE -
testcase_81 AC 1,074 ms
6,944 KB
testcase_82 AC 213 ms
6,940 KB
testcase_83 AC 6 ms
6,944 KB
testcase_84 WA -
testcase_85 AC 4 ms
6,944 KB
testcase_86 AC 28 ms
6,944 KB
testcase_87 AC 18 ms
6,944 KB
testcase_88 AC 17 ms
6,944 KB
testcase_89 AC 1 ms
6,944 KB
testcase_90 AC 1 ms
6,948 KB
testcase_91 AC 1 ms
6,940 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:10:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
   10 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:16:24: note: in expansion of macro 'gc'
   16 |         int n = 0, c = gc();
      |                        ^~

ソースコード

diff # 

// yukicoder: No.738 平らな農地
// 2019.8.4 bal4u

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

typedef long long ll;

#if 1
#define gc() getchar_unlocked()
#else
#define gc() getchar()
#endif

int in() {   // 非負整数の入力
	int n = 0, c = gc();
	do n = 10 * n + (c & 0xf); while ((c = gc()) >= '0');
	return n;
}

// 数値のハッシュ関数
#define HASHSIZ 2000081
typedef struct { int fr, to; } HASH;
HASH hash[HASHSIZ+5], *hashend = hash+HASHSIZ;

int hlookup(int to) {
	HASH *p = hash + to % HASHSIZ;
	while (p->to) {
		if (p->to == to) break;
		if (++p == hashend) p = hash;
	}
	return p->fr;
}

void hinsert(int fr, int to) {
	HASH *p = hash + to % HASHSIZ;
	while (p->to) {
		if (p->to == to) return;
		if (++p == hashend) p = hash;
	}
	p->fr = fr, p->to = to;
}

// データ圧縮
#define MAX 100005
typedef struct { int a, id; } T;
T t[MAX];
int a[MAX], b[MAX]; int N;

int cmp(const void *a, const void *b) { return ((T *)a)->a - ((T *)b)->a; }

int compact(int n, int *to, int *fr) {
	int i, v;
	for (i = 0; i < n; i++) t[i].a = fr[i], t[i].id = i;
	qsort(t, n, sizeof(T), cmp);
	v = 1; to[t[0].id] = v;
	hinsert(t[0].a, v);
	for (i = 1; i < n; i++) {
		if (t[i].a != t[i-1].a) {
			v++;
			to[t[i].id] = v;
			hinsert(t[i].a, v);
		} else to[t[i].id] = v;
	}
	return v;
}

// bit木
int bit[MAX], bitn, bitp;

void init(int maxVal) {
	bitn = maxVal;
	bitp = 1; while (bitp < bitn) bitp <<= 1;
}

void add(int i, int x) {
	i++; while (i <= bitn) bit[i] += x, i += i & -i;
}

void insert(int val) { add(val, 1); }
void erase(int val) { add(val, -1); }
int nth_element(int n) {
	int a = 0, q = bitp;
	while (q >>= 1)
    if (a+q <= bitn && bit[a+q] <= n) a += q, n -= bit[a];
	return a;
}

#define ABS(x)  ((x)>=0?(x):-(x))

ll median(int m, int l, int r) {
	ll s = 0; while (l <= r) s += ABS(a[l]-m), l++;
	return s;
}

int main()
{
	int i, k, K, m;
	ll ans, t;

	N = in(), K = in();
	if (K == 1) { puts("0"); return 0; }
	if (K == 2) {
		m = in(), ans = 0x7fffffff;
		while (--N) {
			k = m, m = in();
			if ((t = ABS(k-m)) < ans) ans = t;
		}
		goto done;
	}
	
	k = K >> 1;
	for (i = 0; i < N; i++) a[i] = in();
	init(compact(N, b, a));
	
	for (i = 0; i < K; i++) insert(b[i]);
	m = hlookup(nth_element(k));
	ans = median(m, 0, K-1);
	if (ans == 0) goto done;
	
	t = ans;
	for (i = K; i < N; i++) {
		erase(b[i-K]), insert(b[i]);
		m = hlookup(nth_element(k));
		t = median(m, i-K+1, i);
		if (t < ans) {
			ans = t;
			if (ans == 0) goto done;
		}
	}
done:
	printf("%lld\n", ans);
	return 0;
}
0