結果

問題 No.1140 EXPotentiaLLL!
ユーザー tonyu0tonyu0
提出日時 2020-07-31 22:34:55
言語 Rust
(1.77.0 + proconio)
結果
WA  
実行時間 -
コード長 1,304 bytes
コンパイル時間 12,156 ms
コンパイル使用メモリ 387,424 KB
実行使用メモリ 23,868 KB
最終ジャッジ日時 2024-07-06 19:47:40
合計ジャッジ時間 16,008 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 225 ms
23,688 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 31 ms
6,940 KB
testcase_09 AC 31 ms
6,944 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 32 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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, (p - 1) / 2 + 1, p)).ok();
        } else {
            writeln!(out, "-1").ok();
        }
    }
    stdout().write_all(&out).unwrap();
}
0