結果
問題 |
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 = 0 for i in range(N): if B[i] == 0: heappush(zero,i) elif B[i] == 1: pass else: 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)