A7716 | Sum of Square of Sum
时间限制2s
内存限制1024MB
通过 / 提交0/0
题目描述
有 $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$
* 所有输入值均为整数。
C++ 编辑器
输入
输出
可保存默认模板；新题优先使用已保存模板。
当前快捷键仅展示，暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好，我是作答助手。可以问思路、复杂度、样例含义或代码报错原因；不会直接给出完整 AC 代码。
确定要清空代码吗？