A7716. Sum of Square of Sum
编程题
普及/提高-
知识点
题目描述
有 $N$ 个编号为 $1$ 到 $N$ 的球。球 $i$ 上写有一个整数 $A_i$。
对于从这 $N$ 个球中任选若干个球的一种方案,定义该方案的**得分**为所选球上所写数字之和的平方。
求从 $N$ 个球中任选 $K$ 个球的所有 $\binom{N}{K}$ 种方案的得分之和,并对 $998244353$ 取模。
对于从这 $N$ 个球中任选若干个球的一种方案,定义该方案的**得分**为所选球上所写数字之和的平方。
求从 $N$ 个球中任选 $K$ 个球的所有 $\binom{N}{K}$ 种方案的得分之和,并对 $998244353$ 取模。
输入格式
输入从标准输入中按以下格式给出:
> $N$ $K$
> $A_1$ $\dots$ $A_N$
> $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$
* 所有输入值均为整数。
从三个球中任选两个球,共有三种选法:选择球 $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$
* 所有输入值均为整数。