結果
問題 | No.59 鉄道の旅 |
ユーザー | szkhts |
提出日時 | 2024-03-17 08:29:21 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 459 bytes |
コンパイル時間 | 350 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 52,464 KB |
最終ジャッジ日時 | 2024-09-30 04:39:20 |
合計ジャッジ時間 | 8,111 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | WA * 4 |
other | TLE * 1 |
ソースコード
n, k = map(int, input().split()) weights = [0] * 1000001 ans = 0 for i in range(n): w = int(input()) tmp = 0 if w < 0: if weights[abs(w)] > 0: weights[abs(w)] -= 1 else: if i < k: weights[i] += 1 else: tmp = sum(weights[i:]) if tmp < k: weights[w] += 1 for i in range(len(weights)): ans += weights[i] print(ans)