#890594 (Rust) No.2324 Two Countries within UEC

• 問題: No.2324 Two Countries within UEC
• ユーザー: kenichi
• 提出日時: 2023-07-14 17:25:03
• 言語: Rust (1.83.0 + proconio)
• 結果: WA
• 実行時間: -
• コード長: 7,861 bytes
• コンパイル時間: 12,746 ms
• コンパイル使用メモリ: 378,576 KB
• 実行使用メモリ: 5,504 KB
• 最終ジャッジ日時: 2024-09-16 02:18:30
• 合計ジャッジ時間: 15,739 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

ファイルパターン	結果
sample	AC * 1 WA * 1
other	AC * 7 WA * 34

コンパイルメッセージ

warning: unused variable: `n`
  --> src/main.rs:17:9
   |
17 |         n: u64,
   |         ^
   |
help: `n` is captured in macro and introduced a unused variable
  --> src/main.rs:88:13
   |
16 | /     input! {
17 | |         n: u64,
18 | |         m: u64,
19 | |         p: u64,
20 | |         q: usize,
21 | |         ask: [(u64, u64); q],
22 | |     }
   | |_____- in this macro invocation
...
88 |           let $var = read_value!($iter, $t);
   |               ^^^^
   = note: `#[warn(unused_variables)]` on by default
   = note: this warning originates in the macro `input_inner` which comes from the expansion of the macro `input` (in Nightly builds, run with -Z macro-backtrace for more info)

ソースコード

use std::io::Write;
//use std::io::*;
//use std::str::FromStr;
/*
//use std::num;
//use std::cmp;
//use std::collections::VecDeque;
//use std::io::{self, BufRead};
const LOW_CASE_ALPHABET: [char; 26] = [
    'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's',
    't', 'u', 'v', 'w', 'x', 'y', 'z',
];
*/

fn run() {
    input! {
        n: u64,
        m: u64,
        p: u64,
        q: usize,
        ask: [(u64, u64); q],
    }
    let pow = |mut r: u64, mut n: u64| -> u64{
        let mut t = 1 % p;
        while n > 0 {
            if n & 1 == 1{
                t = t * r % p;
            }
            r = r * r % p;
            n >>= 1;
        }
        t
    };
    let inv = |a: u64| {
        let a = a % p;
        assert!(a != 0);
        pow(a, p - 2)
    };
    let out = std::io::stdout();
    let mut out = std::io::BufWriter::new(out.lock());
    for (x, f) in ask {
        let x = x % p;
        let ans = if x == 0{
            if f == 0{
                m
            } else {
                0
            }
        } else {
            let y = f * inv(x) % p;
            let mut res = m / p;
            if y > 0 && y <= m * p{
                res += 1;
            }
            res
        };
        writeln!(out, "{}", ans).ok();
    }
}

fn main(){
    run();
}

#[macro_export]
macro_rules! input {
    (source = $s:expr, $($r:tt)*) => {
        let mut iter = $s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
    ($($r:tt)*) => {
        let s = {
            use std::io::Read;
            let mut s = String::new();
            std::io::stdin().read_to_string(&mut s).unwrap();
            s
        };
        let mut iter = s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
}

#[macro_export]
macro_rules! input_inner {
    ($iter:expr) => {};
    ($iter:expr, ) => {};
    ($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($iter, $t);
        input_inner!{$iter $($r)*}
    };
}

#[macro_export]
macro_rules! read_value {
    ($iter:expr, ( $($t:tt),* )) => {
        ( $(read_value!($iter, $t)),* )
    };
    ($iter:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
    };
    ($iter:expr, chars) => {
        read_value!($iter, String).chars().collect::<Vec<char>>()
    };
    ($iter:expr, bytes) => {
        read_value!($iter, String).bytes().collect::<Vec<u8>>()
    };
    ($iter:expr, usize1) => {
        read_value!($iter, usize) - 1
    };
    ($iter:expr, $t:ty) => {
        $iter.next().unwrap().parse::<$t>().expect("Parse error")
    };
}

/*
let mut temp = String::new();
    std::io::stdin().read_line(&mut temp).ok();
    let temp = temp
        .trim()
        .chars()
        .map(|c| c as usize - '0' as usize)
        .collect::<Vec<_>>();

    let mut iter = temp.iter(); // イテレータを取得

    let a: isize = *iter.next().unwrap() as isize;
    let b: isize = *iter.next().unwrap() as isize;
    let c: isize = *iter.next().unwrap() as isize;

*/
/*
pub fn quicksort(numbers: &mut Vec<u64>) {
    if numbers.len() <= 1 {
        return;
    }

    let pivot = numbers.pop().unwrap();
    let mut lesser = Vec::new();
    let mut greater = Vec::new();

    for num in numbers.iter() {
        if *num >= pivot {
            lesser.push(*num);
        } else {
            greater.push(*num);
        }
    }

    quicksort(&mut lesser);
    quicksort(&mut greater);

    numbers.clear();
    numbers.extend(lesser);
    numbers.push(pivot);
    numbers.extend(greater);
}

pub fn is_leap_year(year: isize) -> bool {
    if ((year % 4 == 0) && (year % 100 != 0)) || (year % 400 == 0) {
        return true;
    } else {
        return false;
    }
}

pub fn ctoi(c: char) -> isize {
    let n = c as isize - '0' as isize;

    return n;
}

pub fn stoi(c: String) -> isize {
    let n = c.parse::<isize>().unwrap();

    return n;
}

pub fn find_char(sentence: Vec<char>, target: char, start: isize) -> isize {
    for _i in start as usize..sentence.len() {
        if sentence[_i] == target {
            return _i as isize;
        }
    }
    return -1;
}

pub fn hex_to_binary(hex_string: &str, x: u32) -> Option<String> {
    match i64::from_str_radix(hex_string, x) {
        Ok(decimal) => Some(format!("{}", decimal)),
        Err(_) => None,
    }
}

pub fn my_floor(x: f64) -> f64 {
    let rounded = (x * 1000.0).floor() / 1000.0;

    rounded
}

pub fn upper_lower(x: char) -> char {
    let result: char;

    if x.is_uppercase() {
        result = x.to_ascii_lowercase();
    } else if x.is_lowercase() {
        result = x.to_ascii_uppercase();
    } else {
        result = x;
    }

    result
}

pub fn maxf_2(x: f64, y: f64) -> f64 {
    let result;

    if x >= y {
        result = x;
    } else {
        result = y;
    }

    result
}

pub fn max_2(x: isize, y: isize) -> isize {
    let result;

    if x >= y {
        result = x;
    } else {
        result = y;
    }

    result
}

pub fn max_u_2(x: usize, y: usize) -> usize {
    let result;

    if x >= y {
        result = x;
    } else {
        result = y;
    }

    result
}

pub fn min_u_2(x: usize, y: usize) -> usize {
    let result;

    if x <= y {
        result = x;
    } else {
        result = y;
    }

    result
}

pub fn min_2(x: isize, y: isize) -> isize {
    let result;

    if x >= y {
        result = y;
    } else {
        result = x;
    }

    result
}

pub fn min_2_i32(x: i32, y: i32) -> i32 {
    let result;

    if x >= y {
        result = y;
    } else {
        result = x;
    }

    result
}

pub fn max_3(x: isize, y: isize, z: isize) -> isize {
    let result;
    if x >= y && y >= z {
        result = x;
    } else if x >= z && z >= y {
        result = x;
    } else if y >= x && x >= z {
        result = y;
    } else if y >= z && z >= x {
        result = y;
    } else if z >= x && x >= y {
        result = z;
    } else {
        result = z;
    }

    result
}

pub fn max_u_3(x: usize, y: usize, z: usize) -> usize {
    let result;
    if x >= y && y >= z {
        result = x;
    } else if x >= z && z >= y {
        result = x;
    } else if y >= x && x >= z {
        result = y;
    } else if y >= z && z >= x {
        result = y;
    } else if z >= x && x >= y {
        result = z;
    } else {
        result = z;
    }

    result
}

pub fn min_3(x: isize, y: isize, z: isize) -> isize {
    let result;
    if x <= y && y <= z {
        result = x;
    } else if x <= z && z <= y {
        result = x;
    } else if y <= x && x <= z {
        result = y;
    } else if y <= z && z <= x {
        result = y;
    } else if z <= x && x <= y {
        result = z;
    } else {
        result = z;
    }

    result
}

pub fn min_u_3(x: usize, y: usize, z: usize) -> usize {
    let result;
    if x <= y && y <= z {
        result = x;
    } else if x <= z && z <= y {
        result = x;
    } else if y <= x && x <= z {
        result = y;
    } else if y <= z && z <= x {
        result = y;
    } else if z <= x && x <= y {
        result = z;
    } else {
        result = z;
    }

    result
}

pub fn read_text() -> String {
    let mut word = String::new();
    std::io::stdin().read_line(&mut word).ok();
    let answer = word.trim().to_string();

    answer
}

pub fn read_array() -> Vec<char> {
    let array: String = read();
    let chars: Vec<char> = array.trim().chars().collect::<Vec<char>>();
    chars
}

pub fn read<T: FromStr>() -> T {
    let stdin = stdin();
    let stdin = stdin.lock();
    let token: String = stdin
        .bytes()
        .map(|c| c.expect("failed to read char") as char)
        .skip_while(|c| c.is_whitespace())
        .take_while(|c| !c.is_whitespace())
        .collect();
    token.parse().ok().expect("failed to parse token")
}
*/