結果

問題 No.1318 ABCD quadruplets
コンテスト
ユーザー akakimidori
提出日時 2020-12-15 02:59:16
言語 Rust
(1.94.0 + proconio + num + itertools)
コンパイル:
/usr/bin/rustc_custom
実行:
./target/release/main
結果
AC  
実行時間 134 ms / 2,000 ms
コード長 1,281 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,017 ms
コンパイル使用メモリ 198,528 KB
実行使用メモリ 7,976 KB
最終ジャッジ日時 2026-04-08 13:44:37
合計ジャッジ時間 6,724 ms
ジャッジサーバーID
(参考情報)
judge1_1 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

fn run() {
    let (n, m) = {
        let mut s = String::new();
        std::io::stdin().read_line(&mut s).unwrap();
        let mut it = s.trim().split_whitespace();
        let n: usize = it.next().unwrap().parse().unwrap();
        let m: usize = it.next().unwrap().parse().unwrap();
        (n, m)
    };
    let mut dp = vec![0u16; n + 1];
    let way = [24, 12, 12, 4, 12, 6, 4, 1];
    for d in 0..=m {
        let v = d * d;
        let s = d;
        for c in (0..=d).take_while(|c| c * c + c * s <= n - v) {
            let v = c * c + c * s + v;
            let s = c + s;
            let y = (c == d) as usize;
            for b in (0..=c).take_while(|b| b * b + b * s <= n - v) {
                let v = b * b + b * s + v;
                let s = b + s;
                let y = 2 * y + (b == c) as usize;
                for a in (0..=b).take_while(|a| a * a + a * s <= n - v) {
                    let x = a * a + a * s + v;
                    let y = 2 * y + (a == b) as usize;
                    dp[x] += way[y];
                }
            }
        }
    }
    use std::io::Write;
    let out = std::io::stdout();
    let mut out = std::io::BufWriter::new(out.lock());
    for v in dp {
        writeln!(out, "{}", v).ok();
    }
}

fn main() {
    run();
}
0