結果

問題 No.1972 Modulo Set
ユーザー cutmdocutmdo
提出日時 2022-06-10 23:00:50
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 16 ms / 2,000 ms
コード長 2,085 bytes
コンパイル時間 16,894 ms
コンパイル使用メモリ 379,788 KB
実行使用メモリ 7,368 KB
最終ジャッジ日時 2024-10-05 01:37:10
合計ジャッジ時間 13,120 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

use std::collections::HashMap;
use std::io::Write;
/* ----------------------------------------------- */
fn run<W: Write>(sc: &mut scanner::Scanner, out: &mut std::io::BufWriter<W>) {
    let n: usize = sc.next();
    let m: usize = sc.next();
    let v: Vec<usize> = sc.next_vec(n);

    let mut map: HashMap<usize, usize> = HashMap::new();
    for x in v {
        let p = map.entry(x % m).or_insert(0);
        *p += 1
    }
    let mut ans: usize = 0;
    for (&x, &val) in &map {
        ans += if x == 0 || (x == m >> 1 && m & 1 == 0) {
            std::cmp::min(1, val)
        } else {
            let val_r = *map.get(&(m - x)).unwrap_or(&0);
            if val > val_r || (val == val_r && x < m - x) {
                val
            } else {
                0
            }
        };
    }
    writeln!(out, "{:?}", ans).ok();
}

/* ----------------------------------------------- */
#[allow(dead_code)]
mod scanner {
    use std::str::FromStr;
    pub struct Scanner<'a> {
        it: std::str::SplitWhitespace<'a>,
    }
    impl<'a> Scanner<'a> {
        pub fn new(s: &'a String) -> Scanner<'a> {
            Scanner {
                it: s.split_whitespace(),
            }
        }
        pub fn next<T: FromStr>(&mut self) -> T {
            self.it.next().unwrap().parse::<T>().ok().unwrap()
        }
        pub fn next_bytes(&mut self) -> Vec<u8> {
            self.it.next().unwrap().bytes().collect()
        }
        pub fn next_chars(&mut self) -> Vec<char> {
            self.it.next().unwrap().chars().collect()
        }
        pub fn next_vec<T: FromStr>(&mut self, len: usize) -> Vec<T> {
            (0..len).map(|_| self.next()).collect()
        }
    }
}
/* ----------------------------------------------- */
fn main() {
    use std::io::Read;
    let mut s = String::new();
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut sc = scanner::Scanner::new(&s);
    let out = std::io::stdout();
    let mut out = std::io::BufWriter::new(out.lock());
    run(&mut sc, &mut out);
}
/* ----------------------------------------------- */
0