結果

問題 No.1140 EXPotentiaLLL!
コンテスト
ユーザー tonyu0
提出日時 2020-07-31 22:29:03
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,300 bytes
コンパイル時間 11,297 ms
コンパイル使用メモリ 397,568 KB
実行使用メモリ 24,552 KB
最終ジャッジ日時 2024-07-06 19:36:30
合計ジャッジ時間 14,999 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

use std::io::*;

fn mod_pow(mut x: u64, mut e: u64, m: u64) -> u64 {
    let mut res = 1;
    while e > 0 {
        if e & 1 == 1 {
            res = res * x % m;
        }
        x = x * x % m;
        e >>= 1;
    }
    res
}

fn main() {
    let mut s: String = String::new();
    std::io::stdin().read_to_string(&mut s).ok();
    let mut itr = s.trim().split_whitespace();
    let t: usize = itr.next().unwrap().parse().unwrap();
    const MAX: usize = 5000005;
    let mut is_prime: Vec<bool> = vec![true; MAX];
    is_prime[0] = false;
    is_prime[1] = false;
    for i in 2..MAX {
        if is_prime[i] {
            let mut j = i * 2;
            while j < MAX {
                is_prime[j] = false;
                j += i;
            }
        }
    }

    let mut out = Vec::new();
    for _ in 0..t {
        let a: u64 = itr.next().unwrap().parse().unwrap();
        let p: u64 = itr.next().unwrap().parse().unwrap();

        //
        if is_prime[p as usize] {
            // a^(p!) mod p
            // a^(p - 1) = 1 mod p
            // p * (p -1) * (p -2) * ...Copy
            // x^p == x mod p
            writeln!(out, "{}", mod_pow(a, (p - 1) / 2 + 1, p)).ok();
        } else {
            writeln!(out, "-1").ok();
        }
    }
    stdout().write_all(&out).unwrap();
}
0