AT_arc190_d [ARC190D] Matrix Pow Sum

题目描述

给定素数 $p$ 和一个 $N \times N$ 的矩阵 $A = (A_{i,j})$($1 \leq i,j \leq N$),其中每个元素 $A_{i,j}$ 是 $0$ 到 $p-1$ 之间的整数。 定义矩阵 $B$ 为将 $A$ 中所有 $0$ 替换为 $1$ 到 $p-1$ 之间的整数后得到的矩阵。设 $A$ 中 $0$ 的个数为 $K$,则共有 $(p-1)^K$ 种可能的 $B$。 要求计算所有可能的 $B$ 对应的 $B^p$(矩阵的 $p$ 次幂)之和,并将每个元素对 $p$ 取模后输出。

输入格式

输入格式如下: > $N$ $p$ > $A_{1,1}$ $\cdots$ $A_{1,N}$ > $\vdots$ > $A_{N,1}$ $\cdots$ $A_{N,N}$

输出格式

输出 $N$ 行,每行包含 $N$ 个整数。第 $i$ 行第 $j$ 列表示所有 $B^p$ 之和的 $(i,j)$ 元素对 $p$ 取模后的结果,整数间以空格分隔。

说明/提示

### 约束条件 - $1 \leq N \leq 100$ - $p$ 是满足 $1 \leq p \leq 10^9$ 的素数 - $0 \leq A_{i,j} \leq p-1$ - 输入均为整数 ### 样例解释 1 所有可能的 $B$ 对应的 $B^3$ 如下: - $\begin{pmatrix}1 & 1 \\ 1 & 2\end{pmatrix}^3 = \begin{pmatrix}5 & 8 \\ 8 & 13\end{pmatrix}$ - $\begin{pmatrix}1 & 1 \\ 2 & 2\end{pmatrix}^3 = \begin{pmatrix}9 & 9 \\ 18 & 18\end{pmatrix}$ - $\begin{pmatrix}2 & 1 \\ 1 & 2\end{pmatrix}^3 = \begin{pmatrix}14 & 13 \\ 13 & 14\end{pmatrix}$ - $\begin{pmatrix}2 & 1 \\ 2 & 2\end{pmatrix}^3 = \begin{pmatrix}20 & 14 \\ 28 & 20\end{pmatrix}$ 总和为 $\begin{pmatrix}48 & 44 \\ 67 & 65\end{pmatrix}$,各元素对 $p=3$ 取模后输出。 ### 样例解释 2 所有可能的 $B$ 对应的 $B^2$ 为: - $\begin{pmatrix}1 & 1 & 1 \\ 1 & 1 & 1 \\ 1 & 1 & 1\end{pmatrix}^2 = \begin{pmatrix}3 & 3 & 3 \\ 3 & 3 & 3 \\ 3 & 3 & 3\end{pmatrix}$ 总和对 $p=2$ 取模后输出。 翻译由 DeepSeek R1 完成