結果

問題 No.1084 積の積
ユーザー tonyu0
提出日時 2020-06-20 00:06:22
言語 Rust
(1.83.0 + proconio)
結果
RE  
実行時間 -
コード長 1,090 bytes
コンパイル時間 13,224 ms
コンパイル使用メモリ 407,336 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-03 16:13:41
合計ジャッジ時間 16,794 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4 RE * 1
other AC * 24 RE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;
const MOD: i64 = 1_000_000_007;

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 n: usize = itr.next().unwrap().parse().unwrap();
    let a: Vec<i64> = (0..n)
        .map(|_| itr.next().unwrap().parse().unwrap())
        .collect();

    let mp = |mut x: i64, mut e: i64| -> i64 {
        let mut res = 1;
        while e > 0 {
            if e & 1 == 1 {
                res = res * x % MOD;
            }
            x = x * x % MOD;
            e >>= 1;
        }
        res
    };

    let mut s = vec![0i64; n + 1];
    let mut r = 0;
    let mut now = 1;
    for l in 0..n {
        while r < n && now * a[r] < 1_000_000_000 {
            now *= a[r];
            r += 1;
        }
        for i in 0..r - l {
            s[l + i] += (r - l - i) as i64;
        }
        if l == r {
            r += 1;
        }
        now /= a[l];
    }

    let mut ans = 1;
    for i in 0..n {
        ans = ans * mp(a[i], s[i]) % MOD;
    }
    println!("{}", ans);
}
0