AT_arc190_d.[ARC190D] Matrix Pow Sum

NOI/NOI+/CTSC

通过率:0%

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

给定素数 pp 和一个 N×NN \times N 的矩阵 A=(Ai,j)A = (A_{i,j})(1≤i,j≤N1 \leq i,j \leq N),其中每个元素 Ai,jA_{i,j} 是 00 到 p−1p-1 之间的整数。

定义矩阵 BB 为将 AA 中所有 00 替换为 11 到 p−1p-1 之间的整数后得到的矩阵。设 AA 中 00 的个数为 KK,则共有 (p−1)K(p-1)^K 种可能的 BB。

要求计算所有可能的 BB 对应的 BpB^p(矩阵的 pp 次幂)之和,并将每个元素对 pp 取模后输出。

输入格式

输入格式如下:

NN pp
A1,1A_{1,1} ⋯\cdots A1,NA_{1,N}
⋮\vdots
AN,1A_{N,1} ⋯\cdots AN,NA_{N,N}

输出格式

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

输入输出样例

  • 输入#1

    2 3
    0 1
    0 2

    输出#1

    0 2
    1 2
  • 输入#2

    3 2
    1 0 0
    0 1 0
    0 0 1

    输出#2

    1 1 1
    1 1 1
    1 1 1
  • 输入#3

    4 13
    0 1 2 0
    3 4 0 5
    0 6 0 7
    8 9 0 0

    输出#3

    8 0 6 5
    11 1 8 5
    8 0 4 12
    8 0 1 9

说明/提示

约束条件

  • 1≤N≤1001 \leq N \leq 100
  • pp 是满足 1≤p≤1091 \leq p \leq 10^9 的素数
  • 0≤Ai,j≤p−10 \leq A_{i,j} \leq p-1
  • 输入均为整数

样例解释 1

所有可能的 BB 对应的 B3B^3 如下:

  • (1112)3=(58813)\begin{pmatrix}1 & 1 \\ 1 & 2\end{pmatrix}^3 = \begin{pmatrix}5 & 8 \\ 8 & 13\end{pmatrix}
  • (1122)3=(991818)\begin{pmatrix}1 & 1 \\ 2 & 2\end{pmatrix}^3 = \begin{pmatrix}9 & 9 \\ 18 & 18\end{pmatrix}
  • (2112)3=(14131314)\begin{pmatrix}2 & 1 \\ 1 & 2\end{pmatrix}^3 = \begin{pmatrix}14 & 13 \\ 13 & 14\end{pmatrix}
  • (2122)3=(20142820)\begin{pmatrix}2 & 1 \\ 2 & 2\end{pmatrix}^3 = \begin{pmatrix}20 & 14 \\ 28 & 20\end{pmatrix}
    总和为 (48446765)\begin{pmatrix}48 & 44 \\ 67 & 65\end{pmatrix},各元素对 p=3p=3 取模后输出。

样例解释 2

所有可能的 BB 对应的 B2B^2 为:

  • (111111111)2=(333333333)\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=2p=2 取模后输出。

翻译由 DeepSeek R1 完成

输入解题思路,AI测评打分。不知道怎么写?

首页