結果
問題 | No.2635 MST on Line++ 2 |
ユーザー |
👑 ![]() |
提出日時 | 2024-02-13 15:11:26 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 150 ms / 2,000 ms |
コード長 | 830 bytes |
コンパイル時間 | 4,423 ms |
コンパイル使用メモリ | 259,504 KB |
最終ジャッジ日時 | 2025-02-19 05:53:58 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h> #include<atcoder/all> using namespace std; using namespace atcoder; using mint=modint998244353; #define rep(i,a,b) for (int i=(int)(a);i<(int)(b);i++) int main() { int N, K; cin >> N >> K; vector<int> A(N + 1), D(N); rep(i, 0, N) cin >> A[i]; sort(A.begin(), A.end()); rep(i, 0, N) D[i] = A[i + 1] - A[i]; vector<mint> fact(N + 1, 1), fact_inv(N + 1, 1); rep(i, 0, N) fact[i + 1] = fact[i] * (i + 1); fact_inv[N] = fact[N].inv(); for(int i = N; i > 0; i--) fact_inv[i - 1] = fact_inv[i] * i; vector<mint> g(N); rep(i, 0, N) g[i] = D[i] * fact[N - i]; auto tmp = convolution(g, fact_inv); mint ans = 0; rep(i, 1, N){ int X = N - min(K, i) - min(K, N - i); ans += tmp[X] * fact[X]; } cout << ans.val() << "\n"; }