結果
問題 | No.990 N×Mマス計算(Kの倍数) |
ユーザー |
|
提出日時 | 2023-01-21 01:48:16 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 1,744 ms / 2,000 ms |
コード長 | 1,551 bytes |
コンパイル時間 | 14,581 ms |
コンパイル使用メモリ | 379,108 KB |
実行使用メモリ | 8,628 KB |
最終ジャッジ日時 | 2024-06-23 13:22:11 |
合計ジャッジ時間 | 17,917 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 19 |
コンパイルメッセージ
warning: unused variable: `m` --> src/main.rs:13:9 | 13 | let m = nmk[1]; | ^ help: if this is intentional, prefix it with an underscore: `_m` | = note: `#[warn(unused_variables)]` on by default
ソースコード
use std::collections::HashMap;fn gcd(a: usize, b: usize) -> usize {if b == 0 { return a; }gcd(b, a%b)}fn main() {let mut nmk = String::new();std::io::stdin().read_line(&mut nmk).ok();let nmk: Vec<usize> = nmk.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();let n = nmk[0];let m = nmk[1];let k = nmk[2];let mut temp = String::new();std::io::stdin().read_line(&mut temp).ok();let temp: Vec<&str> = temp.trim().split_whitespace().collect();let op = temp[0].chars().nth(0).unwrap();let b = temp.into_iter().skip(1).map(|v| v.parse::<usize>().unwrap()).collect::<Vec<_>>();let a = (0..n).map(|_| {let mut temp = String::new();std::io::stdin().read_line(&mut temp).ok();let temp: usize = temp.trim().parse().unwrap();temp}).collect::<Vec<_>>();if op == '+' {let mut bmap = HashMap::new();for &v in b.iter() {*bmap.entry(v % k).or_insert(0usize) += 1;}let result = a.iter().map(|&v| *bmap.get(&((k-v%k)%k)).unwrap_or(&0usize)).sum::<usize>();println!("{}", result);return;}let mut bmap = HashMap::new();for &v in b.iter() {*bmap.entry(gcd(k, v)).or_insert(0usize) += 1;}let mut result = 0usize;for &v in a.iter() {for (&bkey, &bcnt) in bmap.iter() {if bkey * v % k == 0 {result += bcnt;}}}println!("{}", result);}