結果
問題 | No.990 N×Mマス計算(Kの倍数) |
ユーザー |
|
提出日時 | 2023-02-18 05:01:33 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 145 ms / 2,000 ms |
コード長 | 1,774 bytes |
コンパイル時間 | 13,675 ms |
コンパイル使用メモリ | 378,576 KB |
実行使用メモリ | 8,624 KB |
最終ジャッジ日時 | 2024-07-19 18:38:15 |
合計ジャッジ時間 | 16,099 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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 cntmap = HashMap::new();for &v in b.iter() {*cntmap.entry(v%k).or_insert(0usize) += 1;}let mut result = 0usize;for &v in a.iter() {if let Some(&x) = cntmap.get(&((k-v%k)%k)) {result += x;}}println!("{}", result);return;}let mut bmap = HashMap::new();for &v in b.iter() {*bmap.entry(gcd(k, v)).or_insert(0usize) += 1;}let mut amap = HashMap::new();for &v in a.iter() {*amap.entry(gcd(k, v)).or_insert(0usize) += 1;}let mut result = 0usize;for (&aval, &acnt) in amap.iter() {for (&bval, &bcnt) in bmap.iter() {if k == gcd(aval*bval, k) {result += acnt * bcnt;}}}println!("{}", result);}