結果

問題 No.1889 K Consecutive Ks (Hard)
ユーザー 👑 ygussany
提出日時 2022-03-17 18:05:28
言語 C
(gcc 13.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other -- * 22
権限があれば一括ダウンロードができます

ソースコード

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

#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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0