結果
問題 |
No.59 鉄道の旅
|
ユーザー |
![]() |
提出日時 | 2017-08-19 14:20:54 |
言語 | D (dmd 2.109.1) |
結果 |
AC
|
実行時間 | 32 ms / 5,000 ms |
コード長 | 1,267 bytes |
コンパイル時間 | 718 ms |
コンパイル使用メモリ | 106,128 KB |
実行使用メモリ | 8,760 KB |
最終ジャッジ日時 | 2024-06-12 21:42:14 |
合計ジャッジ時間 | 1,660 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
import std.algorithm; import std.array; import std.conv; import std.math; import std.range; import std.stdio; import std.string; import std.typecons; int readint() { return readln.chomp.to!int; } int[] readints() { return readln.split.map!(to!int).array; } void main() { auto nk = readints(); int n = nk[0], k = nk[1]; const MAX_W = 1_000_000; auto bit = new BinaryIndexTree(MAX_W); for (int i = 0; i < n; i++) { auto w = readint(); if (w > 0) { // 荷物を積む if (bit.sum(MAX_W) - bit.sum(w - 1) < k) { // k 個未満なら w を積む bit.add(w, 1); } } else { // 荷物を下ろす if (bit.sum(-w) - bit.sum(-w - 1) >= 1) { // w を下ろす bit.add(-w, -1); } } } int ans = bit.sum(MAX_W); writeln(ans); } class BinaryIndexTree { private int[] _bit; this(int n) { _bit = new int[n + 1]; } int sum(int p) { int s = 0; while (p > 0) { s += _bit[p]; p -= p & -p; } return s; } void add(int p, int x) { while (p < _bit.length) { _bit[p] += x; p += p & -p; } } }