結果

問題 No.1140 EXPotentiaLLL!
ユーザー tonyu0tonyu0
提出日時 2020-07-31 22:29:03
言語 Rust
(1.72.1)
結果
WA  
実行時間 -
コード長 1,300 bytes
コンパイル時間 749 ms
コンパイル使用メモリ 147,616 KB
実行使用メモリ 27,268 KB
最終ジャッジ日時 2023-09-21 00:52:50
合計ジャッジ時間 5,106 ms
ジャッジサーバーID
(参考情報)
judge11 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 110 ms
24,624 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 35 ms
6,784 KB
testcase_09 AC 36 ms
6,888 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 36 ms
6,876 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 - 1) / 2 + 1, p)).ok();
        } else {
            writeln!(out, "-1").ok();
        }
    }
    stdout().write_all(&out).unwrap();
}
0