AT_arc167_c [ARC167C] MST on Line++

题目描述

给定正整数 $N,K$ 以及一个长度为 $N$ 的正整数序列 $A=(A_{1},A_{2},\dots,A_{N})$。 对于 $ (1,2,\dots,N) $ 的一个排列 $P=(P_{1},P_{2},\dots,P_{N})$,我们考虑如下的“问题 MST on Line”,并将其答案记为 $f(P)$。 > **问题 MST on Line** > > 有一个无向带权图 $G$,包含 $N$ 个顶点,顶点编号为 $1$ 到 $N$。对于任意满足 $1\leq i > - 如果 $j-i\leq K$,则在顶点 $i$ 和顶点 $j$ 之间存在一条边,该边的权值为 $\max(A_{P_{i}},A_{P_{j}})$。 > - 如果 $j-i>K$,则顶点 $i$ 和顶点 $j$ 之间没有边。 > > 请计算图 $G$ 的最小生成树的所有边权之和。 请你求出所有 $ (1,2,\dots,N) $ 的排列 $P=(P_{1},P_{2},\dots,P_{N})$ 对应的 $f(P)$ 之和,并对 $998244353$ 取模后输出。

输入格式

输入通过标准输入给出,格式如下: > $N$ $K$ $A_{1}$ $A_{2}$ $\cdots$ $A_{N}$

输出格式

请输出答案。

说明/提示

## 限制条件 - $1\leq K