結果

問題 No.59 鉄道の旅
コンテスト
ユーザー bal4u
提出日時 2019-06-01 22:27:42
言語 C(gnu17)
(gcc 15.2.0)
コンパイル:
gcc-15 -O2 -std=gnu17 -Wno-error=implicit-function-declaration -Wno-error=implicit-int -Wno-error=incompatible-pointer-types -Wno-error=int-conversion -DONLINE_JUDGE -o a.out _filename_ -lm
実行:
./a.out
結果
AC  
実行時間 11 ms / 5,000 ms
コード長 1,126 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 213 ms
コンパイル使用メモリ 39,528 KB
最終ジャッジ日時 2026-02-22 03:31:27
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

// yukicoder: 59 鉄道の旅
// 2019.6.1 bal4u

#include <stdio.h>

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

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

#define MAXVAL 1000005
int bit[MAXVAL], sz, p;
void init(int maxVal) {
	sz = maxVal;
	p = 1; while (p < sz) p <<= 1;
}

void add(int i, int x) {
	i++; while (i <= sz)
		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 = p; n--;
	while (q >>= 1)
		if (a+q <= sz && bit[a+q] <= n)
			a += q, n -= bit[a];
	return a;
}

int f[MAXVAL];

int main()
{
	int N, K, w, ans;
	
	N = in(), K = in(), ans = 0;
	init(MAXVAL);
	while (N--) {
		w = in();
		if (w > 0) {
			if (ans < K || nth_element(K) > MAXVAL-w)
				ans++, f[w]++, insert(MAXVAL-w);
		} else {
			w = -w;
			if (f[w] > 0) f[w]--, ans--, erase(MAXVAL-w);
		}
	}
	printf("%d\n", ans);
	return 0;
}
0