結果
問題 | No.990 N×Mマス計算(Kの倍数) |
ユーザー |
|
提出日時 | 2023-02-04 05:37:11 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 64 ms / 2,000 ms |
コード長 | 1,782 bytes |
コンパイル時間 | 13,610 ms |
コンパイル使用メモリ | 384,312 KB |
実行使用メモリ | 8,632 KB |
最終ジャッジ日時 | 2024-07-03 03:50:51 |
合計ジャッジ時間 | 16,126 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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 mut result = 0usize;for &v in a.iter() {let v = v % k;let target = (k - v) % k;result += *bmap.get(&target).unwrap_or(&0usize);}println!("{}", result);return;}let mut amap = HashMap::new();let mut bmap = HashMap::new();for &v in a.iter() {*amap.entry(gcd(v, k)).or_insert(0usize) += 1;}for &v in b.iter() {*bmap.entry(gcd(k, v)).or_insert(0usize) += 1;}let mut result = 0usize;for (&a, &acnt) in amap.iter() {for (&b, &bcnt) in bmap.iter() {if a * b % k == 0 {result += acnt * bcnt;}}}println!("{}", result);}