結果
問題 | No.929 よくあるボールを移動するやつ |
ユーザー |
|
提出日時 | 2020-12-08 16:13:56 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 254 ms / 2,000 ms |
コード長 | 473 bytes |
コンパイル時間 | 166 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 16,288 KB |
最終ジャッジ日時 | 2024-09-17 14:43:31 |
合計ジャッジ時間 | 2,472 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
from heapq import *N = int(input())B = list(map(int,input().split()))zero = []stock = []count = 0for i in range(N):if B[i] == 0:heappush(zero,i)elif B[i] == 1:passelse:heappush(stock,(i,B[i]))while zero != [] and stock != []:z = heappop(zero)s,i = heappop(stock)if i > 1:heappush(stock,(s,i-1))else:s,i = heappop(stock)heappush(stock,(s,i-1))count += abs(z-s)print(count)