結果

問題 No.1281 Cigarette Distribution
ユーザー Strorkis
提出日時 2020-11-06 23:15:32
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,639 bytes
コンパイル時間 14,157 ms
コンパイル使用メモリ 377,260 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-22 13:37:32
合計ジャッジ時間 13,110 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

pub mod procon_input {
    use std::{str::FromStr, iter::FromIterator};

    pub fn read_line() -> String {
        let mut input = String::new();
        std::io::stdin().read_line(&mut input).ok();
        input
    }

    pub fn parse<T: FromStr>(s: &str) -> T {
        s.parse().ok().unwrap()
    }
    
    pub fn read<T: FromStr>() -> T {
        parse(read_line().trim_end())
    }
    
    pub fn read_collection<T: FromStr, C: FromIterator<T>>() -> C {
        read_line().split_whitespace().map(parse).collect()
    }

    #[macro_export]
    macro_rules! read_tuple {
        ( $( $t:ty ),* ) => {{
            let input = read_line();
            let mut iter = input.split_whitespace();
            ( $( parse::<$t>(iter.next().unwrap()) ),* )
        }};
    }
}

use procon_input::*;

const MOD: i64 = 1_000_000_007;

fn solve(writer: &mut std::io::BufWriter<std::io::StdoutLock>) {
    use std::io::Write;

    let (n, m) = read_tuple!(usize, usize);

    let pow = |mut a: i64, mut n: usize| -> i64 {
        let mut res = 1;
        while n > 0 {
            if n & 1 > 0 {
                res = res * a % MOD;
            }
            a = a * a % MOD;
            n >>= 1;
        }
        res
    };    

    for x in 1..=m {
        if n <= x {
            writeln!(writer, "0").ok();
            continue;
        }

        let q = n / x;
        let r = n % x;
        let ans = pow((q + 1) as i64, r) * pow(q as i64, x - r);
        writeln!(writer, "{}", ans).ok();
    }
}

fn main() {
    let stdout = std::io::stdout();
    let mut writer = std::io::BufWriter::new(stdout.lock());
    solve(&mut writer);
}
0