AT_arc120_f2.[ARC120F2] Wine Thief

NOI/NOI+/CTSC

通过率:0%

AC君温馨提醒

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

题目描述

问题 F 和问题 F2 是相同的问题,但约束条件和运行时间限制不同。

高桥君的仓库里有 NN 瓶葡萄酒,按左右方向排成一列。从左边数第 ii 瓶葡萄酒的美味度为 AiA_i。
青木君现在要从这 NN 瓶葡萄酒中,恰好选出 KK 瓶进行偷窃。但是,高桥君非常警觉,如果满足以下条件,他就会发现被偷了:

  • 存在连续排列的 DD 瓶葡萄酒,其中被偷走的有 22 瓶或以上。

请你求出所有不会被高桥君发现的偷窃方案中,偷走的葡萄酒美味度之和的总和。
由于答案可能非常大,请输出其对 998244353998244353 取模的结果。

输入格式

输入以如下格式从标准输入给出:

NN KK DD A1A_1 A2A_2 A3A_3 …\dots ANA_N

输出格式

输出答案对 998244353998244353 取模的结果。

输入输出样例

  • 输入#1

    4 2 2
    1 4 2 3

    输出#1

    14
  • 输入#2

    5 2 3
    1 5 7 7 3

    输出#2

    20
  • 输入#3

    18 4 4
    107367523 266126484 149762920 57456082 857431610 400422663 768881284 494753774 152155823 740238343 871191740 450057094 208762450 787961742 90197530 77329823 193815114 707323467

    输出#3

    228955567

说明/提示

约束条件

  • 2≤D≤N≤1062 \leq D \leq N \leq 10^6
  • 1≤K≤⌈ND⌉1 \leq K \leq \left\lceil \frac{N}{D} \right\rceil(⌈x⌉\left\lceil x \right\rceil 表示不小于 xx 的最小整数)
  • 1≤Ai<9982443531 \leq A_i < 998244353
  • 输入中的所有值均为整数

样例解释 1

偷窃方案及其美味度之和如下:

  • 偷第 11 瓶和第 33 瓶:美味度之和为 1+2=31 + 2 = 3
  • 偷第 11 瓶和第 44 瓶:美味度之和为 1+3=41 + 3 = 4
  • 偷第 22 瓶和第 44 瓶:美味度之和为 4+3=74 + 3 = 7
    因此答案为 3+4+7=143 + 4 + 7 = 14。

样例解释 2

偷窃方案及其美味度之和如下:

  • 偷第 11 瓶和第 44 瓶:美味度之和为 1+7=81 + 7 = 8
  • 偷第 11 瓶和第 55 瓶:美味度之和为 1+3=41 + 3 = 4
  • 偷第 22 瓶和第 55 瓶:美味度之和为 5+3=85 + 3 = 8

由 ChatGPT 4.1 翻译

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

首页