测评会员优惠活动进行中 · 开通 VIP,有效期内测评不限次 VIP 优惠中 · 测评不限次 立即查看

A7716. Sum of Square of Sum

编程题 普及/提高-

题目描述

有 $N$ 个编号为 $1$ 到 $N$ 的球。球 $i$ 上写有一个整数 $A_i$。

对于从这 $N$ 个球中任选若干个球的一种方案,定义该方案的**得分**为所选球上所写数字之和的平方。

求从 $N$ 个球中任选 $K$ 个球的所有 $\binom{N}{K}$ 种方案的得分之和,并对 $998244353$ 取模。

输入格式

输入从标准输入中按以下格式给出:

> $N$ $K$
> $A_1$ $\dots$ $A_N$

输出格式

输出答案。

输入输出样例

输入 #1
3 2
1 10 100
输出 #1
22422
输入 #2
5 2
10 10 20 20 20
输出 #2
10600
输入 #3
2 1
998244353 998244353
输出 #3
0

说明/提示

**样例 1 解释:**
从三个球中任选两个球,共有三种选法:选择球 $1$ 和 $2$ 得分 $(1+10)^2=121$;选择球 $1$ 和 $3$ 得分 $(1+100)^2=10201$;选择球 $2$ 和 $3$ 得分 $(10+100)^2=12100$。

答案为这三项之和:$121+10201+12100=22422$。

**样例 2 解释:**
多个球上可能写有相同的数字。

**样例 3 解释:**
求和结果对 $998244353$ 取模。

### 约束条件

* $1 \leq K \leq N \leq 2\times 10^5$
* $1 \leq A_i \leq 10^9$
* 所有输入值均为整数。
上一题 去做题 下一题