結果
問題 | No.929 よくあるボールを移動するやつ |
ユーザー |
![]() |
提出日時 | 2020-08-26 21:43:57 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 611 ms / 2,000 ms |
コード長 | 1,095 bytes |
コンパイル時間 | 2,310 ms |
コンパイル使用メモリ | 79,492 KB |
実行使用メモリ | 50,620 KB |
最終ジャッジ日時 | 2024-11-07 13:56:03 |
合計ジャッジ時間 | 9,279 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
import java.util.*;public class Main {public static void main (String[] args) {Scanner sc = new Scanner(System.in);int n = sc.nextInt();TreeMap<Integer, Integer> stock = new TreeMap<>();ArrayDeque<Integer> zeros = new ArrayDeque<>();long total = 0;for (int i = 0; i < n; i++) {int x = sc.nextInt();if (x == 0) {if (stock.size() == 0) {zeros.push(i);} else {int key = stock.lastKey();total += i - key;if (stock.get(key) == 1) {stock.remove(key);} else {stock.put(key, stock.get(key) - 1);}}} else if (x > 1) {for (int j = 1; j < x; j++) {if (zeros.size() == 0) {stock.put(i, x - j);break;} else {total += i - zeros.pop();}}}}System.out.println(total);}}