結果
問題 | No.1505 Zero-Product Ranges |
ユーザー |
|
提出日時 | 2023-12-02 10:26:16 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 13 ms / 2,000 ms |
コード長 | 2,125 bytes |
コンパイル時間 | 11,985 ms |
コンパイル使用メモリ | 380,064 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-26 16:25:57 |
合計ジャッジ時間 | 15,335 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 49 |
ソースコード
#![allow(dead_code, unused_variables, unused_macros, unused_imports)]fn main() {input!(n:usize, a:[u64;n]);let ones: Vec<_> = a.split(|&ai| ai == 0).map(|arr| arr.len()).collect();let triangular = |x| x * (x + 1) / 2;let ans = ones.into_iter().fold(triangular(n), |acc, l| acc - triangular(l));println!("{}", ans);}mod xxx {// https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8#[macro_export]macro_rules! input {(source = $s:expr, $($r:tt)*) => {input_inner!{$s, $($r)*}};($($r:tt)*) => {let stdin = std::io::stdin();let mut bytes = std::io::Read::bytes(std::io::BufReader::new(stdin.lock()));let mut next = move || -> String{bytes.by_ref().map(|r|r.unwrap() as char).skip_while(|c|c.is_whitespace()).take_while(|c|!c.is_whitespace()).collect()};input_inner!{next, $($r)*}};}#[macro_export]macro_rules! input_inner {($next:expr) => {};($next:expr, ) => {};($next:expr, mut $var:ident : $t:tt $($r:tt)*) => {let mut $var = read_value!($next, $t);input_inner!{$next $($r)*}};($next:expr, $var:ident : $t:tt $($r:tt)*) => {let $var = read_value!($next, $t);input_inner!{$next $($r)*}};}#[macro_export]macro_rules! read_value {($next:expr, ( $($t:tt),* )) => {( $(read_value!($next, $t)),* )};($next:expr, [ $t:tt ; $len:expr ]) => {(0..$len).map(|_| read_value!($next, $t)).collect::<Vec<_>>()};($next:expr, chars) => {read_value!($next, String).chars().collect::<Vec<char>>()};($next:expr, usize1) => {read_value!($next, usize) - 1};($next:expr, bits)=>{u64::from_str_radix(&$next(), 2).expect("Parse error")};($next:expr, hexs)=>{u64::from_str_radix(&$next(), 16).expect("Parse error")};($next:expr, $t:ty) => {$next().parse::<$t>().expect("Parse error")};}}