結果

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

ソースコード

diff #
raw source code 

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

MX = 1000000
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:
        W -= 1
        if BIT_Sum(MX - W) < K:
            BIT_Add(MX - W, 1)
    elif W < 0:
        W += 1

        if BIT_Sum(MX + W) - BIT_Sum(MX + W - 1) > 0:
            BIT_Add(MX + W, -1)

print BIT_Sum(MX)
0