#abc471e. Sum of Square of Sum

    ID: 4387 Type: Default 1000ms 256MiB Tried: 0 Accepted: 0 Difficulty: 4 Uploaded By: Tags>普及+/提高组合数学二项式系数恒等式变形

Sum of Square of Sum

题目描述

有编号为 11NNNN 个球,第 ii 个球上写着整数 AiA_i

对于「从 NN 个球中选出若干个球」的一种选法,定义它的得分为:所选球上数字之和的平方

请求出「从 NN 个球中恰好选 KK 个」的全部 (NK)\dbinom{N}{K} 种选法的得分总和,对 998244353998244353 取模。

输入格式

N K
A_1 ... A_N

输出格式

输出答案对 998244353998244353 取模的结果。

输入示例 1

3 2
1 10 100

输出示例 1

22422

示例 1 说明

33 个球中选 22 个共有 33 种选法:

  • 选球 1,21, 2:得分 (1+10)2=121(1 + 10)^2 = 121
  • 选球 1,31, 3:得分 (1+100)2=10201(1 + 100)^2 = 10201
  • 选球 2,32, 3:得分 (10+100)2=12100(10 + 100)^2 = 12100

合计 121+10201+12100=22422121 + 10201 + 12100 = 22422

输入示例 2

5 2
10 10 20 20 20

输出示例 2

10600

示例 2 说明

多个球上可能写着相同的数字,此时它们仍然算作不同的球,选法要分别计数。

输入示例 3

2 1
998244353 998244353

输出示例 3

0

示例 3 说明

两个球上的数都恰好等于模数 998244353998244353,在模意义下都等于 00,所以答案是 00注意 AiA_i 本身可以大于等于模数,读入后要及时取模。

约束条件

  • 1KN2×1051 \le K \le N \le 2 \times 10^5
  • 1Ai1091 \le A_i \le 10^9
  • 所有输入值均为整数