T573807 [语言月赛 202502] 沿轴求和
题目背景
# 本题得分将不再按照时间流逝递减,赛后将统一以满分 1700 分为标准重测。
题目描述
给定一个高维数组 $a$,该数组共有 $n$ 维,大小为 $d_0\times d_1\times \cdots \times d_{n-1}$。其中,每一维称为一个轴,$d_i$ 对应 $i(0 \le i < n)$ 号轴。数组下标从 $0$ 开始。
高维数组沿 $x$ 号轴的求和结果,是一个 $n-1$ 维的数组 $S$,其大小为 $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}]}
$$
现在,给定该高维数组与 $x$,求沿 $x$ 号轴求和的结果。
输入格式
无
输出格式
无
说明/提示
### 样例解释 1
样例 1 给出了一个二维数组,其大小为 $2\times 2$,沿 1 号轴求和。在二维数组中,按 0 号轴求和可理解为按列求和,按 1 号轴求和可理解为按行求和。
$$
\begin{bmatrix}
1 & 2 \\
3 & 4
\end{bmatrix}
\rightarrow
\begin{bmatrix}
3 & 7
\end{bmatrix}
$$
### 样例解释 2
样例 2 给出了一个三维数组,其大小为 $2\times 2\times 2$。沿 0 号轴求和的结果大小为 $2\times 2$。
以 $S[0][0]$ 的计算方式为例:
$$
S[0][0]=\sum\limits_{i=0}^1{a[i][0][0]=a[0][0][0]+a[1][0][0]=1+5=6}
$$
可以得到结果为
$$
\begin{bmatrix}
6 & 8\\
10 & 12
\end{bmatrix}
$$
### 数据规模与约定
- 对于 $60\%$ 的测试数据,保证 $n=2$。
- 对于 $100\%$ 的测试数据,保证 $2 \le n \le 16$,$1 \le d_i \le 10^3$,$1 \le \prod\limits_{i=0}^{n-1}{d_i} \le 2^{16}$,$-100 \le val \le 100$。保证 $\forall i\in [0,n)$,输入的 $p_i \in [0,d_i)$。