結果

問題 No.2857 Div Array
ユーザー InTheBloomInTheBloom
提出日時 2024-08-25 16:35:23
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 92 ms / 2,000 ms
コード長 1,861 bytes
コンパイル時間 5,801 ms
コンパイル使用メモリ 177,896 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-08-25 16:35:31
合計ジャッジ時間 6,317 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import std;
void main () {
const long MOD = 998244353;
int N, M, K; readln.read(N, M, K);
// -> K...
// -> -> ()
// √M ->
long[][] prod (long[][] A, long[][] B) {
const int d = A.length.to!int;
auto res = new long[][](d, d);
foreach (i; 0..d) {
foreach (j; 0..d) {
foreach (k; 0..d) {
res[i][j] += 1L * A[i][k] * B[k][j] % MOD;
}
res[i][j] %= MOD;
}
}
return res;
}
int[int] count;
foreach (i; 1..M + 1) count[M / i]++;
auto res = new int[](0);
foreach (k, v; count) {
res ~= k;
}
res.sort;
const int d = count.length.to!int;
auto mat = new long[][](d, d);
auto vec = new long[](d);
foreach (i; 0..d) vec[i] = count[res[i]];
foreach (i; 0..d) {
foreach (j; 0..d) {
if (abs(res[i] - res[j]) <= K) mat[i][j] = count[res[j]];
}
}
//
auto m = new long[][](d, d);
foreach (i; 0..d) m[i][i] = 1;
N--;
foreach (i; 0..31) {
if (0 < (N & (1L << i))) {
m = prod(mat, m);
}
mat = prod(mat, mat);
}
//
long ans = 0;
foreach (i; 0..d) {
foreach (j; 0..d) {
ans += 1L * vec[i] * m[i][j] % MOD;
}
ans %= MOD;
}
writeln(ans);
}
void read (T...) (string S, ref T args) {
import std.conv : to;
import std.array : split;
auto buf = S.split;
foreach (i, ref arg; args) {
arg = buf[i].to!(typeof(arg));
}
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0