結果
問題 |
No.885 アマリクエリ
|
ユーザー |
|
提出日時 | 2022-11-30 01:41:43 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 244 ms / 2,000 ms |
コード長 | 1,332 bytes |
コンパイル時間 | 14,588 ms |
コンパイル使用メモリ | 377,256 KB |
実行使用メモリ | 10,704 KB |
最終ジャッジ日時 | 2024-10-07 09:04:39 |
合計ジャッジ時間 | 17,038 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
コンパイルメッセージ
warning: unused variable: `q` --> src/main.rs:13:9 | 13 | let q: usize = q.trim().parse().unwrap(); | ^ help: if this is intentional, prefix it with an underscore: `_q` | = note: `#[warn(unused_variables)]` on by default
ソースコード
use std::collections::BTreeMap; fn main() { let mut n = String::new(); std::io::stdin().read_line(&mut n).ok(); let n: usize = n.trim().parse().unwrap(); let mut a = String::new(); std::io::stdin().read_line(&mut a).ok(); let a: Vec<usize> = a.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let mut q = String::new(); std::io::stdin().read_line(&mut q).ok(); let q: usize = q.trim().parse().unwrap(); let mut x = String::new(); std::io::stdin().read_line(&mut x).ok(); let x: Vec<usize> = x.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let mut mapping = BTreeMap::new(); for i in 0..n { *mapping.entry(a[i] % x[0]).or_insert(0usize) += 1; } let mut summary = mapping.iter().map(|(&val, &cnt)| val * cnt).sum::<usize>(); println!("{}", summary); for &v in x.iter().skip(1) { let stack = mapping.range(v..).map(|(&key, &val)| (key, val)).collect::<Vec<_>>(); summary -= stack.iter().map(|&(val, cnt)| val * cnt).sum::<usize>(); summary += stack.iter().map(|&(val, cnt)| (val % v) * cnt).sum::<usize>(); println!("{}", summary); for &(val, cnt) in stack.iter() { mapping.remove(&val); *mapping.entry(val%v).or_insert(0usize) += cnt; } } }