AT_abc399_f.[ABC399F] Range Power Sum

提高+/省选-

通过率:0%

AC君温馨提醒

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

题目描述

给定正整数 N,KN, K 以及长度为 NN 的整数序列 A=(A1,A2,…,AN)A = (A_1, A_2, \dots, A_N)。
请计算以下表达式的值对 998244353998244353 取模后的余数:

∑1≤l≤r≤N(∑l≤i≤rAi)K\displaystyle \sum_{1 \leq l \leq r \leq N} \left( \sum_{l \leq i \leq r} A_i \right)^K

输入格式

输入通过标准输入给出,格式如下:

NN KK
A1A_1 A2A_2 …\dots ANA_N

输出格式

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

输入输出样例

  • 输入#1

    3 2
    3 1 2

    输出#1

    75
  • 输入#2

    1 10
    0

    输出#2

    0
  • 输入#3

    10 5
    91 59 85 60 57 72 12 3 27 16

    输出#3

    428633385

说明/提示

约束条件

  • 1≤N≤2×1051 \leq N \leq 2 \times 10^5
  • 1≤K≤101 \leq K \leq 10
  • 0≤Ai<9982443530 \leq A_i < 998244353
  • 输入均为整数

样例解释 1

所求的值为 A12+A22+A32+(A1+A2)2+(A2+A3)2+(A1+A2+A3)2=32+12+22+42+32+62=75A_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

注意需要对结果取 998244353998244353 的模。

翻译由 DeepSeek R1 完成

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

首页