結果
問題 |
No.59 鉄道の旅
|
ユーザー |
|
提出日時 | 2022-01-24 21:46:33 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 174 ms / 5,000 ms |
コード長 | 1,035 bytes |
コンパイル時間 | 192 ms |
コンパイル使用メモリ | 82,272 KB |
実行使用メモリ | 93,592 KB |
最終ジャッジ日時 | 2024-12-15 00:35:10 |
合計ジャッジ時間 | 2,682 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
import sys def input(): return sys.stdin.readline().rstrip() class Seg: def __init__(self, N): self.N = 1 << (N.bit_length()) self.arr = [0]*(2*self.N) def set(self, i, v): i += self.N self.arr[i] = v while i > 1: i //= 2 self.arr[i] = self.arr[2*i] + self.arr[2*i+1] def get(self, a, b): a += self.N b += self.N ans = 0 while a <= b: if a % 2 == 1: ans += self.arr[a] a += 1 if b % 2 == 0: ans += self.arr[b] b -= 1 a //= 2 b //= 2 return ans def main(): N, K = map(int, input().split()) W = [int(input()) for i in range(N)] MAX = 1000000 seg = Seg(MAX) for i in W: if i < 0: seg.set(-i, max(0, seg.get(-i, -i)-1)) elif seg.get(i, MAX) < K: seg.set(i, seg.get(i, i)+1) print(seg.get(1, MAX)) if __name__ == '__main__': main()