#B4147. [语言月赛 202502] 沿轴求和
[语言月赛 202502] 沿轴求和
题目描述
给定一个高维数组 ,该数组共有 维,大小为 。其中,每一维称为一个轴, 对应 号轴。数组下标从 开始。
高维数组沿 号轴的求和结果,是一个 维的数组 ,其大小为 $d_0\times d_1 \times \cdots \times d_{x-1} \times d_{x+1} \times \cdots \times d_{n-1}$。其中,
$$S[w_0][w_1]\cdots[w_{x-1}][w_{x+1}]\cdots[w_{n-1}]=\sum\limits_{i=0}^{d_x-1}{a[w_0][w_1]\cdots[w_{x-1}][i][w_{x+1}]\cdots[w_{n-1}]} $$现在,给定该高维数组与 ,求沿 号轴求和的结果。
输入格式
第一行为两个整数 。
第二行为 个整数,依次表示 。
接下来 行,每行 个整数 ,表示 。
输出格式
输出 $\dfrac{d_0\times d_1\times \cdots \times d_{n-1}}{d_x}$ 行,每行 个整数 ,表示 $S[p_0][p_1]\cdots[p_{x-1}][p_{x+1}]\cdots[p_{n-1}]=w$。
你可以以任意顺序输出这些行。
2 1
2 2
0 0 1
0 1 2
1 1 4
1 0 3
1 7
0 3
3 0
2 2 2
0 0 0 1
0 0 1 2
0 1 0 3
0 1 1 4
1 0 0 5
1 0 1 6
1 1 0 7
1 1 1 8
0 0 6
0 1 8
1 0 10
1 1 12
提示
样例解释 1
样例 1 给出了一个二维数组,其大小为 ,沿 1 号轴求和。在二维数组中,按 0 号轴求和可理解为按列求和,按 1 号轴求和可理解为按行求和。
$$\begin{bmatrix} 1 & 2 \\ 3 & 4 \end{bmatrix} \rightarrow \begin{bmatrix} 3 & 7 \end{bmatrix} $$样例解释 2
样例 2 给出了一个三维数组,其大小为 。沿 0 号轴求和的结果大小为 。
以 的计算方式为例:
$$S[0][0]=\sum\limits_{i=0}^1{a[i][0][0]=a[0][0][0]+a[1][0][0]=1+5=6} $$可以得到结果为
数据规模与约定
- 对于 的测试数据,保证 。
- 对于 的测试数据,保证 ,,,。保证 ,输入的 。