結果

問題 No.1578 A × B × C
ユーザー Masaki KitaguchiMasaki Kitaguchi
提出日時 2022-01-13 18:35:14
言語 Rust
(1.77.0)
結果
AC  
実行時間 1 ms / 2,000 ms
コード長 2,006 bytes
コンパイル時間 12,232 ms
コンパイル使用メモリ 378,680 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-04-28 16:09:00
合計ジャッジ時間 13,504 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,820 KB
testcase_02 AC 1 ms
6,816 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 1 ms
6,944 KB
testcase_05 AC 1 ms
6,944 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 1 ms
6,940 KB
testcase_08 AC 1 ms
6,944 KB
testcase_09 AC 1 ms
6,944 KB
testcase_10 AC 1 ms
6,940 KB
testcase_11 AC 0 ms
6,944 KB
testcase_12 AC 1 ms
6,948 KB
testcase_13 AC 1 ms
6,940 KB
testcase_14 AC 1 ms
6,944 KB
testcase_15 AC 1 ms
6,944 KB
testcase_16 AC 1 ms
6,944 KB
testcase_17 AC 1 ms
6,940 KB
testcase_18 AC 1 ms
6,940 KB
testcase_19 AC 1 ms
6,940 KB
testcase_20 AC 1 ms
6,940 KB
testcase_21 AC 1 ms
6,944 KB
testcase_22 AC 1 ms
6,944 KB
testcase_23 AC 1 ms
6,940 KB
testcase_24 AC 1 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[macro_export]
macro_rules! setup {
    { mut $input:ident: SplitWhitespace $(,)? } => {
        use std::io::Read;
        let mut buf = String::new();
        std::io::stdin().read_to_string(&mut buf).unwrap();
        let mut $input = buf.split_whitespace();
    };
}

#[macro_export]
macro_rules! parse_next {
    ($str_iter:expr) => {
        $str_iter.next().unwrap().parse().unwrap()
    };
}

fn main() {
    setup! { mut input: SplitWhitespace };

    let a: u64 = parse_next!(input);
    let b: u64 = parse_next!(input);
    let c: u64 = parse_next!(input);
    let k: u64 = parse_next!(input);

    const MOD: u64 = 1000000007;

    macro_rules! b {
        ($i:expr, +, $j:expr, $mod:expr) => {
            ($i + $j) % $mod
        };
        ($i:expr, -, $j:expr, $mod:expr) => {{
            ($i + $mod - $j) % $mod
        }};
        ($i:expr, *, $j:expr, $mod:expr) => {
            ($i * $j) % $mod
        };
        ($i:expr, ^, $j:expr, $mod:expr) => {{
            let mut a = $i;
            let mut n = $j;
            let mut power = 1;
            while n > 0 {
                if n & 1 == 1 {
                    power = b!(power, *, a, $mod);
                }
                a = b!(a, *, a, $mod);
                n >>= 1;
            }
            power
        }};
    }

    let abc = b!(b!(a, *, b, MOD), *, c, MOD);

    if abc == 0 {
        println!("{}", 0);
        return;
    }

    // ans = (abc ^ {2 ^ k}) % MOD
    // (abc ^ {MOD - 1}) % MOD = 1 (if abc % MOD != 0)
    // (2 ^ k) = (MOD - 1) * ((2 ^ k) / (MOD - 1)) + ((2 ^ k) % (MOD - 1))
    // ans = (abc ^ {(MOD - 1) * ((2 ^ k) / (MOD - 1)) + ((2 ^ k) % (MOD - 1))}) % MOD
    //     = (((abc ^ {MOD - 1}) ^ {(2 ^ k) / (MOD - 1)}) * (abc ^ {(2 ^ k) % (MOD - 1)})) % MOD
    //     = ((((abc ^ {MOD - 1}) % MOD) ^ {(2 ^ k) / (MOD - 1)}) * (abc ^ {(2 ^ k) % (MOD - 1)})) % MOD
    //     = (abc ^ {(2 ^ k) % (MOD - 1)}) % MOD

    let ans = b!(abc, ^, b!(2, ^, k, MOD - 1), MOD);

    println!("{}", ans);
}
0