AT_abc471_e.Sum of Square of Sum

普及/提高-

通过率:0%

时间限制:2.00s

内存限制:1024MB

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

NN 个编号为 11NN 的球。球 ii 上写有一个整数 AiA_i

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

求从 NN 个球中任选 KK 个球的所有 (NK)\binom{N}{K} 种方案的得分之和,并对 998244353998244353 取模。

输入格式

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

NN KK
A1A_1 \dots ANA_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 解释:
从三个球中任选两个球,共有三种选法:选择球 1122 得分 (1+10)2=121(1+10)^2=121;选择球 1133 得分 (1+100)2=10201(1+100)^2=10201;选择球 2233 得分 (10+100)2=12100(10+100)^2=12100

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

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

样例 3 解释:
求和结果对 998244353998244353 取模。

约束条件

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

输入解题思路,AI测评打分。不知道怎么写?

首页