AT_arc120_f [ARC120F] Wine Thief

题目描述

[problemUrl]: https://atcoder.jp/contests/arc120/tasks/arc120_f **问题 F 和问题 F2 是相同的问题,但约束条件和运行时间限制不同。** 高桥君的仓库里有 $N$ 瓶葡萄酒,按左右方向排成一行。从左起第 $i$ 瓶葡萄酒的美味度为 $A_i$。 青木君现在要从这 $N$ 瓶葡萄酒中,恰好选出 $K$ 瓶来偷走。但是,高桥君非常警觉,如果满足以下条件,他就会发现葡萄酒被偷了: - 存在某一段连续的 $D$ 瓶葡萄酒,其中被偷走的有 $2$ 瓶或以上(本题中 $D=2$)。 请计算在不被高桥君发现的所有偷盗方式中,偷走的葡萄酒美味度之和的总和。 由于答案可能非常大,请输出其对 $998244353$ 取模的结果。

输入格式

输入以如下格式从标准输入给出: > $N$ $K$ $D$ $A_1$ $A_2$ $A_3$ $\dots$ $A_N$

输出格式

请输出答案对 $998244353$ 取模的结果。

说明/提示

## 约束条件 - $D=2$ - $2 \leq N \leq 3 \times 10^5$ - $1 \leq K \leq \left\lceil \frac{N}{D} \right\rceil$($\left\lceil x \right\rceil$ 表示大于等于 $x$ 的最小整数) - $1 \leq A_i < 998244353$ - 输入中的所有数值均为整数 ## 样例解释 1 偷盗方式及偷走葡萄酒美味度之和如下: - 偷第 $1$ 瓶和第 $3$ 瓶:美味度之和为 $1+2=3$ - 偷第 $1$ 瓶和第 $4$ 瓶:美味度之和为 $1+3=4$ - 偷第 $2$ 瓶和第 $4$ 瓶:美味度之和为 $4+3=7$ 因此答案为 $3+4+7=14$。 ## 样例解释 2 只能偷第 $1,3,5$ 瓶葡萄酒。 由 ChatGPT 4.1 翻译