結果

問題 No.59 鉄道の旅
コンテスト
ユーザー yuki2006
提出日時 2014-11-05 22:37:07
言語 PyPy2
(7.3.20)
結果
RE  
実行時間 -
コード長 540 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 265 ms
コンパイル使用メモリ 77,188 KB
最終ジャッジ日時 2025-12-03 12:55:37
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 3 RE * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N, K = map(int, raw_input().split())

MX = 10000
data = [0] * (MX + 1)


def BIT_Add(x, a):
    while x <= MX:
        data[x] += a
        x += x & (-x)


def BIT_Sum(x):
    total = 0
    while x > 0:
        total += data[x]
        x -= x & (-x)
    return total


for _ in xrange(N):
    W = int(raw_input())
    if W > 0:
        if BIT_Sum(MX - W) < K:
            BIT_Add(MX - W, 1)
    elif W < 0:
        if BIT_Sum(MX + W) - BIT_Sum(MX + W - 1) > 0:
            BIT_Add(MX + W, -1)

print BIT_Sum(MX)
0