結果
問題 | No.929 よくあるボールを移動するやつ |
ユーザー |
![]() |
提出日時 | 2019-11-22 22:32:59 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 1,232 bytes |
コンパイル時間 | 16,357 ms |
コンパイル使用メモリ | 380,812 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-10-11 04:15:32 |
合計ジャッジ時間 | 14,459 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
#![allow(unused_imports)]#![allow(non_snake_case)]use std::cmp::*;use std::collections::*;use std::io::Write;#[allow(unused_macros)]macro_rules! debug {($($e:expr),*) => {#[cfg(debug_assertions)]$({let (e, mut err) = (stringify!($e), std::io::stderr());writeln!(err, "{} = {:?}", e, $e).unwrap()})*};}fn main() {let n = read::<usize>();let v = read_vec::<usize>();let mut zero_poses = vec![];let mut count = VecDeque::new();for i in 0..n {if v[i] == 0 {zero_poses.push(i);}if v[i] > 1 {count.push_back((v[i] - 1, i));}}let mut ans = 0;for pos in zero_poses {ans += (pos as i64 - count[0].1 as i64).abs();count[0].0 -= 1;if count[0].0 == 0 {count.pop_front();}}println!("{}", ans);}fn read<T: std::str::FromStr>() -> T {let mut s = String::new();std::io::stdin().read_line(&mut s).ok();s.trim().parse().ok().unwrap()}fn read_vec<T: std::str::FromStr>() -> Vec<T> {read::<String>().split_whitespace().map(|e| e.parse().ok().unwrap()).collect()}