結果
問題 |
No.59 鉄道の旅
|
ユーザー |
👑 |
提出日時 | 2022-07-04 17:09:00 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 174 ms / 5,000 ms |
コード長 | 1,167 bytes |
コンパイル時間 | 227 ms |
コンパイル使用メモリ | 82,468 KB |
実行使用メモリ | 84,352 KB |
最終ジャッジ日時 | 2024-12-14 08:23:15 |
合計ジャッジ時間 | 2,731 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
class Bit: def __init__(self, n): self.size = n self.n0 = 1 << (n.bit_length() - 1) self.tree = [0] * (n + 1) def range_sum(self, l, r): return self.sum(r - 1) - self.sum(l - 1) def sum(self, i): i += 1 s = 0 while i > 0: s += self.tree[i] i -= i & -i return s def get(self, i): return self.sum(i) - self.sum(i - 1) def add(self, i, x): i += 1 while i <= self.size: self.tree[i] += x i += i & -i def lower_bound(self, x): pos = 0 plus = self.n0 while plus > 0: if pos + plus <= self.size and self.tree[pos + plus] < x: x -= self.tree[pos + plus] pos += plus plus //= 2 return pos n, k = map(int, input().split()) m = 10 ** 6 + 1 bit = Bit(m) for _ in range(n): w = int(input()) if w > 0: cnt = bit.range_sum(w, m) if cnt < k: bit.add(w, 1) else: w *= -1 if bit.get(w) >= 1: bit.add(w, -1) print(bit.range_sum(0, m))