結果
問題 | No.346 チワワ数え上げ問題 |
ユーザー |
![]() |
提出日時 | 2020-04-13 13:18:39 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 3 ms / 2,000 ms |
コード長 | 1,432 bytes |
コンパイル時間 | 11,915 ms |
コンパイル使用メモリ | 402,264 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-09-24 18:50:45 |
合計ジャッジ時間 | 13,225 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
#![allow(unused_imports)] #![allow(non_snake_case)] use std::collections::HashMap; use std::collections::HashSet; #[allow(unused_macros)] macro_rules! read { ([$t:ty] ; $n:expr) => ((0..$n).map(|_| read!([$t])).collect::<Vec<_>>()); ($($t:ty),+ ; $n:expr) => ((0..$n).map(|_| read!($($t),+)).collect::<Vec<_>>()); ([$t:ty]) => (rl().split_whitespace().map(|w| w.parse().unwrap()).collect::<Vec<$t>>()); ($t:ty) => (rl().parse::<$t>().unwrap()); ($($t:ty),*) => {{ let buf = rl(); let mut w = buf.split_whitespace(); ($(w.next().unwrap().parse::<$t>().unwrap()),*) }}; } #[allow(dead_code)] fn rl() -> String { let mut buf = String::new(); std::io::stdin().read_line(&mut buf).unwrap(); buf.trim_end().to_owned() } fn main() { let s = read!(String); let cw: Vec<char> = s.chars().filter(|c| c == &'c' || c == &'w').collect(); let mut table_w = Vec::<usize>::with_capacity(cw.len()); for _ in 0..cw.len() { table_w.push(0); } for i in (0..cw.len() as isize-1).rev() { let i = i as usize; let dif = if cw[i+1] == 'w' {1} else {0}; table_w[i] = table_w[i+1] + dif; } let mut acc = 0u64; for (i, c) in cw.iter().enumerate() { if c == &'c' { acc += ((table_w[i] as i64) * (table_w[i] as i64 - 1) / 2) as u64; } } println!("{}", acc); }