結果
問題 |
No.929 よくあるボールを移動するやつ
|
ユーザー |
|
提出日時 | 2020-09-03 13:03:33 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 280 ms / 2,000 ms |
コード長 | 682 bytes |
コンパイル時間 | 188 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 12,648 KB |
最終ジャッジ日時 | 2024-11-23 05:39:13 |
合計ジャッジ時間 | 2,679 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
from collections import deque n = int(input()) b = list(map(int, input().split())) have = deque() ans = 0 r = 0 for i in range(n): if b[i] > 1: have.append((i, b[i])) if b[i] == 0: if have != deque(): g, p = have.popleft() b[i] += 1 b[g] -= 1 ans += i-g if p-1 > 1: have.appendleft((g, p-1)) else: if b[r] > 1: g = r else: for g in range(i, n): if b[g] > 1: r = g break b[i] += 1 b[g] -= 1 ans += g-i print(ans)