AT_KeioPC2025_k Loaded Dice

题目描述

けむにく君想用 $N$ 个失真的骰子来复现一个公平的 $K$ 面骰子。 请计算满足以下条件的有理数构成的 $N$ 行 $K$ 列矩阵 $P = (P_{i,j})_{0 \le i < N,\ 0 \le j < K}$ 的个数,并对 $998244353$ 取模后输出。 - 对所有 $(i,j)$,有 $0 \le P_{i,j} \le 1$。 - 对所有 $i$,有 $\sum_{j=0}^{K-1} P_{i,j} = 1$。 - 对所有满足 $0 \le S < K$ 的整数 $S$,有 $$ \sum_{\substack{0 \le j_0, j_1, \dots, j_{N-1} < K \\[0.5mm] j_0 + j_1 + \dots + j_{N-1} = S}} \prod_{0 \le i < N} P_{i,j_i} = \frac{1}{K} $$

输入格式

输入为一行,包括两个整数: $N$ $K$

输出格式

输出答案。

说明/提示

### 样例解释 1 $P = \begin{pmatrix} 1 & 0 \\ \frac{1}{2} & \frac{1}{2} \\ \end{pmatrix},\begin{pmatrix} \frac{1}{2} & \frac{1}{2} \\ 1 & 0 \\ \end{pmatrix}$ 满足条件。 ### 数据范围 - $1 \le N \le 2 \times 10^5$ - $1 \le K \le 2 \times 10^5$ - 输入均为整数。 由 ChatGPT 5 翻译