結果
| 問題 | No.1889 K Consecutive Ks (Hard) |
| ユーザー |
 ygussany
|
| 提出日時 | 2022-03-17 18:05:28 |
| 言語 | C (gcc 12.3.0) |
| 結果 |
TLE
(最新)
AC
(最初)
|
| 実行時間 | - |
| コード長 | 942 bytes |
| コンパイル時間 | 383 ms |
| コンパイル使用メモリ | 33,792 KB |
| 実行使用メモリ | 13,640 KB |
| 最終ジャッジ日時 | 2024-10-04 07:32:04 |
| 合計ジャッジ時間 | 7,652 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge5 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 1 ms
6,816 KB |
| testcase_01 | AC | 1 ms
6,816 KB |
| testcase_02 | TLE | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
ソースコード
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <stdio.h>
#define N_MAX 200000
#define M_MAX 200000
const int Mod = 998244353;
long long pow_mod(int n, long long k)
{
long long N, ans = 1;
for (N = n; k > 0; k >>= 1, N = N * N % Mod) if (k & 1) ans = ans * N % Mod;
return ans;
}
int naive(int N, int M)
{
int i, j, d[N_MAX + 1] = {};
long long dp[N_MAX + 1], tmp, sum = 0;
for (i = 2; i <= M; i++) for (j = i; j <= N; j += i) d[j]++;
for (i = 1, dp[0] = 1; i <= N; i++) {
for (j = 0, tmp = 0; j < i - 1; j++) tmp += dp[j] * d[i-j];
tmp %= Mod;
dp[i] = pow_mod(M - 1, i) - (sum + tmp) % Mod;
if (dp[i] >= Mod) dp[i] -= Mod;
else if (dp[i] < 0) dp[i] += Mod;
sum = (sum * (M - 1) + tmp * (M - 2)) % Mod;
}
return (pow_mod(M, N) - dp[N] % Mod + Mod) % Mod;
}
int main()
{
int N, M;
scanf("%d %d", &N, &M);
printf("%d\n", naive(N, M));
fflush(stdout);
return 0;
}