結果
| 問題 | No.59 鉄道の旅 | 
| コンテスト | |
| ユーザー |  | 
| 提出日時 | 2015-02-24 16:23:11 | 
| 言語 | PyPy2 (7.3.15) | 
| 結果 | 
                                AC
                                 
                             | 
| 実行時間 | 441 ms / 5,000 ms | 
| コード長 | 507 bytes | 
| コンパイル時間 | 242 ms | 
| コンパイル使用メモリ | 76,928 KB | 
| 実行使用メモリ | 87,680 KB | 
| 最終ジャッジ日時 | 2024-12-24 20:32:09 | 
| 合計ジャッジ時間 | 4,018 ms | 
| ジャッジサーバーID (参考情報) | judge1 / judge5 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 12 | 
ソースコード
import math
N,K = map(int,raw_input().split())
W = [int(raw_input()) for i in xrange(N)]
mw = 1000001
have = [0]*mw
D = int(math.ceil(mw**0.5))
bucket = [0]*(mw/D+1)
for i in xrange(N):
    p = abs(W[i])/D
    if W[i] > 0:
        cnt = 0
        for j in xrange(p,mw/D+1): cnt += bucket[j]
        for j in xrange(p*D,W[i]): cnt -= have[j]
        if cnt < K:
            have[W[i]] += 1
            bucket[p] += 1
    elif have[-W[i]] > 0:
        have[-W[i]] -= 1
        bucket[p] -= 1
print sum(bucket)
            
            
            
        