結果

問題 No.1140 EXPotentiaLLL!
ユーザー ikdikd
提出日時 2020-07-31 22:04:39
言語 Rust
(1.83.0 + proconio)
結果
RE  
実行時間 -
コード長 1,199 bytes
コンパイル時間 25,103 ms
コンパイル使用メモリ 384,624 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-06 18:12:43
合計ジャッジ時間 14,612 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 1 RE * 8 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::Read;

fn read<T: std::str::FromStr>() -> T {
    let token: String = std::io::stdin()
        .bytes()
        .map(|c| c.ok().unwrap() as char)
        .skip_while(|c| c.is_whitespace())
        .take_while(|c| !c.is_whitespace())
        .collect();
    token.parse().ok().unwrap()
}

fn mpow(a: i64, x: i64, mo: i64) -> i64 {
    if x == 0 {
        1
    } else if x == 1 {
        a
    } else if (x & 1) == 1 {
        a * mpow(a, x - 1, mo) % mo
    } else {
        mpow(a * a % mo, x / 2, mo) % mo
    }
}

fn main() {
    let n = 5 * 1000 + 1;
    let mut is_prime = vec![true; n];
    is_prime[1] = false;
    for i in 2..n {
        if i * i >= n {
            break;
        }
        if is_prime[i] {
            let mut j = i * 2;
            while j < n {
                is_prime[j] = false;
                j += i;
            }
        }
    }

    let t: usize = read();
    for _ in 0..t {
        let a: i64 = read();
        let p: i64 = read();
        let ans = if p == 2 {
            a * a % p
        } else if is_prime[p as usize] {
            mpow(a, p * p / 2 % (p - 1), p)
        } else {
            -1
        };
        println!("{}", ans);
    }
}
0