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