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)$。