AT_abc471_e [ABC471E] Sum of Square of Sum

题目描述

现有编号为 $1$ 到 $N$ 的 $N$ 个球,第 $i$ 个球上面写着一个整数 $A_i$。 现在从这 $N$ 个球中选取若干个球,其得分定义为所选球上的数字之和的平方。 $N$ 个球里恰好选取 $K$ 个球有 $\binom{N}{K}$ 种方案,请计算每一种方案的答案的总和,结果对 $998244353$ 取模。

输入格式

第一行两个整数 $N,K$。 第二行 $N$ 个整数 $A_1,A_2,A_3,\cdots,A_N$。

输出格式

输出答案。

说明/提示

### 样例解释一 从 $3$ 个球中选取 $2$ 个球的方案总共有 $3$ 种。 - 选取球 $1,2$ 时,得分为 $(1+10)^2=121$。 - 选取球 $1,3$ 时,得分为 $(1+100)^2=10201$。 - 选取球 $2,3$ 时,得分为 $(10+100)^2=12100$。 所求答案为得分之和:$121+10201+12100=22422$。 ### 样例解释二 多个球上可能写着相同的数。 ### 样例解释三 请输出对 $998244353$ 取模后的结果。 ### 数据范围 - $1 \le K \le N \le 2 \times 10^5$ - $1 \le A_i \le 10^9$ - 输入均为整数