結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2020-03-06 22:25:18 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 8 ms / 1,500 ms |
コード長 | 1,259 bytes |
コンパイル時間 | 13,183 ms |
コンパイル使用メモリ | 378,088 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-14 08:01:47 |
合計ジャッジ時間 | 13,984 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#![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 v = read_vec::<usize>(); let (n, k) = (v[0], v[1] - 1); let a = read_vec::<i64>(); if a[k] == 0 { println!("0"); return; } let mut right = 0i64; for i in k + 1..n { right += a[i]; if a[i] <= 1 { break; } } let mut left = 0i64; for i in (0..k).rev() { left += a[i]; if a[i] <= 1 { break; } } if a[k] == 1 { println!("{}", max(left, right) + 1); } else { println!("{}", left + right + a[k]); } } 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() }