class RangeCountSegTree(object): '''以下の操作をサポートするデータ構造を提供する。 1) 数値の追加、削減 2) ある数値の要素がいくつあるかの問合せ 3) ある数値以上の要素がいくつあるかの問合せ 数値は、0..size-1 の範囲内とする。 (飛び飛びの値の場合は、全データをrankにおきかえてから使えばよい。) 同じ数値を何度でも追加することができる。 self.data : 子ノードに含まれる個数を格納した segment tree ''' def __init__(self, size): self.data = [0] * (2 * size) self.size = size def __getitem__(self, val): return self.data[self.size + val] def add(self, val): '''値 val を持った要素を追加する。 ''' pos = self.size + val data = self.data while pos: data[pos] += 1 pos >>= 1 def sub(self, val): pos = self.size + val data = self.data while pos: data[pos] -= 1 pos >>= 1 def count_more_than_or_equal(self, val): data = self.data pos = self.size + val count = data[pos] while pos: if not pos & 1: count += data[pos+1] pos >>= 1 return count if __name__ == '__main__': N, K = map(int, input().split()) Ws = [] Wset = set() for n in range(N): w = int(input()) if w > 0: Ws.append(w) Wset.add(w) elif -w in Wset: Ws.append(w) uniqW = list(Wset) uniqW.sort() val2idx = {w: i for i, w in enumerate(uniqW)} size = 1 << (len(bin(len(uniqW))) - 2) seg = RangeCountSegTree(size) for w in Ws: if w > 0: v = val2idx[w] if seg.count_more_than_or_equal(v) >= K: continue seg.add(v) else: v = val2idx[-w] if seg[v] <= 0: continue seg.sub(v) print(seg.count_more_than_or_equal(0))