結果
問題 | No.1470 Mex Sum |
ユーザー |
|
提出日時 | 2021-04-19 20:08:00 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 25 ms / 2,000 ms |
コード長 | 1,577 bytes |
コンパイル時間 | 15,680 ms |
コンパイル使用メモリ | 382,768 KB |
実行使用メモリ | 14,768 KB |
最終ジャッジ日時 | 2024-07-04 05:03:05 |
合計ジャッジ時間 | 15,424 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
use std::collections::*; #[allow(unused)] struct Scanner { stack: VecDeque<String>, input: std::io::Stdin, } impl Scanner { fn new() -> Scanner { Scanner { stack: VecDeque::<String>::new(), input: std::io::stdin(), } } fn take(&mut self) -> Option<String> { if let Some(s) = self.stack.pop_front() { return Some(s); } self.enqueue_from_line(); self.stack.pop_front() } fn enqueue_from_line(&mut self) { let mut t = String::new(); self.input.read_line(&mut t).ok(); t.trim() .split_ascii_whitespace() .for_each(|s| self.stack.push_back(s.to_owned())); } } macro_rules! scan { ($io:expr => $t:ty) => { $io.take().unwrap().parse::<$t>().unwrap() }; ($io:expr => $t:tt * $n:expr) => ((0..$n).map(|_| scan!($io => $t)).collect::<Vec<_>>()); ($io:expr => $($t:tt),*) => (($(scan!($io => $t)),*)); } fn main() { let mut cin = Scanner::new(); let n = scan!(cin => i32); let aa = scan!(cin => i32 * n); let mut one = 0 as i64; let mut two = 0 as i64; let mut three = 0 as i64; for a in aa { match a { 1 => one += 1, 2 => two += 1, _ => three += 1, } } let mut total = 0; total += 2 * one * (one - 1) / 2; total += 1 * two * (two - 1) / 2; total += 1 * three * (three - 1) / 2; total += 3 * one * two; total += 2 * one * three; total += 1 * two * three; println!("{}", total); }