結果
| 問題 | No.59 鉄道の旅 | 
| コンテスト | |
| ユーザー |  | 
| 提出日時 | 2015-02-24 15:44:47 | 
| 言語 | PyPy2 (7.3.15) | 
| 結果 | 
                                WA
                                 
                             | 
| 実行時間 | - | 
| コード長 | 529 bytes | 
| コンパイル時間 | 347 ms | 
| コンパイル使用メモリ | 76,928 KB | 
| 実行使用メモリ | 87,680 KB | 
| 最終ジャッジ日時 | 2024-12-24 20:31:15 | 
| 合計ジャッジ時間 | 4,580 ms | 
| ジャッジサーバーID (参考情報) | judge1 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 10 WA * 2 | 
ソースコード
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(W[i],min(mw,(p+1)*D)): cnt += have[j]
        for j in xrange(p+1,mw/D):     cnt += bucket[j]
        if cnt < K:
            have[W[i]] += 1
            bucket[p] += 1
    elif have[abs(W[i])] > 0:
        have[abs(W[i])] -= 1
        bucket[p] -= 1
print sum(have)
            
            
            
        