結果

問題 No.942 プレゼント配り
ユーザー noisy_noiminnoisy_noimin
提出日時 2019-12-07 21:32:39
言語 Rust
(1.77.0)
結果
WA  
実行時間 -
コード長 1,497 bytes
コンパイル時間 1,747 ms
コンパイル使用メモリ 141,960 KB
実行使用メモリ 12,928 KB
最終ジャッジ日時 2023-08-27 03:10:19
合計ジャッジ時間 3,755 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 20 ms
4,380 KB
testcase_02 AC 16 ms
12,928 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 AC 17 ms
12,904 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 1 ms
4,380 KB
testcase_12 AC 2 ms
4,380 KB
testcase_13 AC 1 ms
4,380 KB
testcase_14 WA -
testcase_15 AC 12 ms
4,376 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 1 ms
4,380 KB
testcase_19 AC 1 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

use std::io::*;
use std::str::FromStr;

fn read<T: FromStr>() -> T {
    let stdin = stdin();
    let stdin = stdin.lock();
    let token: String = stdin
        .bytes()
        .map(|c| c.expect("failed to read char") as char)
        .skip_while(|c| c.is_whitespace())
        .take_while(|c| !c.is_whitespace())
        .collect();
    token.parse().ok().expect("failed to parse token")
}

fn main() {
    let n: usize = read();
    let k: usize = read();
    let m = n / k;  // 一人当たりのプレゼントの数
    let mut ans = vec![vec![0; m]; k];

    if ((k+1) * m) % k != 0 {
        println!("No");
        return;
    }

    if m % 2 == 0 {
        for i in 0..m {
            for j in 0..k {
                if i % 2 == 0 {
                    ans[j][i] = (i+1) * k - j
                } else {
                    ans[j][i] = i * k + j + 1
                }
            }
        }
    } else {
        let mut d = 0;
        for dd in 1..k {
            if (dd * m) % k == 0 {
                d = dd;
                break;
            }
        }

        if d == 0 {
            println!("No");
            return;
        }

        for i in 0..m {
            let diff = (i * d) % k;
            for j in 0..k {
                ans[(j+diff)%k][i] = i * k + j + 1;
            }
        }
    }

    println!("Yes");
    for i in 0..k {
        print!("{}", ans[i][0]);
        for j in 1..m {
            print!(" {}", ans[i][j]);
        }
        print!("\n");
    }
}
0