No.2635 MST on Line++ 2
レベル : / 実行時間制限 : 1ケース 2.000秒 / メモリ制限
: 512 MB / 標準ジャッジ問題
タグ : / 解いたユーザー数 13
作問者 : 👑 potato167 / テスター : ebi_fly noya2
タグ : / 解いたユーザー数 13
作問者 : 👑 potato167 / テスター : ebi_fly noya2
問題文最終更新日: 2024-02-16 00:25:25
問題文
正整数 $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 2」について考え、その答えを $f(P)$ と書きます。
問題 MST on Line 2
頂点に $1$ から $N$ までの番号がついた頂点数 $N$ の重み付き無向グラフ $G$ があります。$G$ について $1\leq i\lt j\leq N$ を満たす任意の整数の組 $(i,j)$ に対して以下が成り立ちます。
- $j-i\leq K$ ならば頂点 $i$ と頂点 $j$ の間に辺が存在して、その辺の重みは $\min(A_{P_{i}},A_{P_{j}})$
- $j-i\gt K$ ならば頂点 $i$ と頂点 $j$ の間に辺は存在しない
$G$ の最小全域木の辺の重みの和を求めてください。
全ての $(1,2,\dots ,N)$ の順列 $P=(P_{1},P_{2},\dots ,P_{N})$ に対する $f(P)$ の総和を $998244353$ で割ったあまりを求めてください。
制約
- $1\leq K\lt N\leq$ $2\times 10^{5}$
- $1\leq A_{i}\leq 10^{9}$
- 入力は全て整数
入力
$N$ $K$ $A_{1}$ $A_{2}$ $\cdots$ $A_{N}$
出力
答えを出力してください。
サンプル
サンプル1
入力
5 2 3 4 5 2 1
出力
648
サンプル2
入力
2 1 924 167
出力
334
サンプル3
入力
12 9 22847 98332 854 68844 81080 46058 40949 62493 76561 52907 88628 99740
出力
475260609
サンプル4
入力
17 3 20508 7568 87729 35274 46341 50587 23669 42805 25381 3806 28055 80697 94519 33016 90880 370 81146
出力
92588775
提出するには、Twitter 、GitHub、 Googleもしくは右上の雲マークをクリックしてアカウントを作成してください。