AT_abc471_e.Sum of Square of Sum
普及/提高-
通过率:0%
时间限制:2.00s
内存限制:1024MB
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
有 N 个编号为 1 到 N 的球。球 i 上写有一个整数 Ai。
对于从这 N 个球中任选若干个球的一种方案,定义该方案的得分为所选球上所写数字之和的平方。
求从 N 个球中任选 K 个球的所有 (KN) 种方案的得分之和,并对 998244353 取模。
输入格式
输入从标准输入中按以下格式给出:
N K
A1 … AN
输出格式
输出答案。
输入输出样例
输入#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≤K≤N≤2×105
- 1≤Ai≤109
- 所有输入值均为整数。
输入解题思路,AI测评打分。不知道怎么写?