結果
問題 |
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); } }