結果

問題 No.1140 EXPotentiaLLL!
ユーザー ikdikd
提出日時 2020-07-31 22:11:33
言語 Rust
(1.77.0 + proconio)
結果
WA  
実行時間 -
コード長 1,187 bytes
コンパイル時間 15,687 ms
コンパイル使用メモリ 379,240 KB
実行使用メモリ 7,040 KB
最終ジャッジ日時 2024-07-06 18:27:28
合計ジャッジ時間 22,586 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

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)
    }
}

fn main() {
    let n = 5 * 1000000 + 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 / 2 + 1, p)
        } else {
            -1
        };
        println!("{}", ans);
    }
}
0