結果
問題 | No.929 よくあるボールを移動するやつ |
ユーザー |
|
提出日時 | 2020-12-08 16:12:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 245 ms / 2,000 ms |
コード長 | 473 bytes |
コンパイル時間 | 130 ms |
コンパイル使用メモリ | 82,324 KB |
実行使用メモリ | 94,996 KB |
最終ジャッジ日時 | 2024-09-17 14:43:27 |
合計ジャッジ時間 | 2,612 ms |
ジャッジサーバーID (参考情報) |
judge6 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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)