AT_abc399_f [ABC399F] Range Power Sum

题目描述

[problemUrl]: https://atcoder.jp/contests/abc399/tasks/abc399_f 给定正整数 $N, K$ 以及长度为 $N$ 的整数序列 $A = (A_1, A_2, \dots, A_N)$。 请计算以下表达式的值对 $998244353$ 取模后的余数: $$ \displaystyle \sum_{1 \leq l \leq r \leq N} \left( \sum_{l \leq i \leq r} A_i \right)^K $$

输入格式

输入通过标准输入给出,格式如下: > $N$ $K$ > $A_1$ $A_2$ $\dots$ $A_N$

输出格式

输出计算结果对 $998244353$ 取模后的余数。

说明/提示

### 约束条件 - $1 \leq N \leq 2 \times 10^5$ - $1 \leq K \leq 10$ - $0 \leq A_i < 998244353$ - 输入均为整数 ### 样例解释 1 所求的值为 $A_1^2 + A_2^2 + A_3^2 + (A_1 + A_2)^2 + (A_2 + A_3)^2 + (A_1 + A_2 + A_3)^2 = 3^2 + 1^2 + 2^2 + 4^2 + 3^2 + 6^2 = 75$。 ### 样例解释 3 注意需要对结果取 $998244353$ 的模。 翻译由 DeepSeek R1 完成