luogu#P8590. 『JROI-8』这是新历的朝阳,也是旧历的残阳

    ID: 12575 远端评测题 1000ms 128MiB 尝试: 0 已通过: 0 难度: 4 上传者: 标签>洛谷原创前缀和洛谷月赛双指针,two-pointer

『JROI-8』这是新历的朝阳,也是旧历的残阳

题目背景

1663764375173.png

少女于海边伫立,凝视着落日最后的余晖
“已然过去了呢,旧历的一年......”

已获得转载授权。

题目描述

给定序列 {an}\{a_n\},满足每一项都不小于前一项。对于所有不超过 kk 的正整数 mm,询问如果将 aa 分成 mm 段(可以有空段),并给从前往后第 ii 段内的每个数都加上 ii,增加后的 j=1naj2\sum\limits_{j=1}^n a_j^2 最大是多少。询问相互独立,即每次询问时给每个数加的值不保留到下一次询问。

例如,对于序列 {3,1,2,2}\{-3,1,2,2\},若 m=5m=5,则一种分段方式是 [3][][1,2][][2][-3][][1,2][][2],增加后的序列是 2,4,5,7-2,4,5,7,此时 j=1naj2=94\sum\limits_{j=1}^n a_j^2=94

m=im=i 时的答案(即此时最大的 j=1naj2\sum\limits_{j=1}^n a_j^2)为 qiq_i,出于良心考虑,你只需要输出 $\left(\sum\limits_{i=1}^k q_i\right) \bmod 998244353$ 即可。标准程序不基于特殊的输出方式,即能独立求出每一个 qiq_i

输入格式

第一行两个正整数 n,kn,k,同题意。

接下来一行 nn 个整数,表示 {an}\{a_n\}

输出格式

一行一个整数,表示 $\left(\sum\limits_{i=1}^k q_i\right) \bmod 998244353$。

4 3
-3 1 2 2
141

提示

【样例解释】

m=1m=1 时,最优策略是 [3,1,2,2][-3,1,2,2]q1=(2)2+22+32+32=26q_1=(-2)^2+2^2+3^2+3^2=26

m=2m=2 时,最优策略是 [3][1,2,2][-3][1,2,2]q2=(2)2+32+42+42=45q_2=(-2)^2+3^2+4^2+4^2=45

m=3m=3 时,最优策略是 [3][][1,2,2][-3][][1,2,2]q3=(2)2+42+52+52=70q_3=(-2)^2+4^2+5^2+5^2=70

则 $\left(\sum\limits_{i=1}^k q_i\right) \bmod 998244353=(q_1+q_2+q_3)\bmod 998244353=(26+45+70)\bmod 998244353=141$。

【数据范围与约束】

测试点编号 分数 nn\leq kk\leq ai\lvert a_i\rvert \leq 特殊性质
131\sim 3 1515 1212 10001000
464\sim 6 10001000
787\sim 8 1010 10610^6 10610^6 10710^7 ai0a_i\geq0
9129 \sim 12 2020 10001000
132013\sim 20 4040 10710^7