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$
- 输入均为整数