結果

問題 No.1887 K Consecutive Ks (Easy)
ユーザー 👑 ygussany
提出日時 2022-03-07 15:45:13
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 136 ms / 2,000 ms
コード長 899 bytes
コンパイル時間 1,159 ms
コンパイル使用メモリ 30,080 KB
実行使用メモリ 72,192 KB
最終ジャッジ日時 2024-07-22 10:40:16
合計ジャッジ時間 2,180 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

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

#include <stdio.h>
const int Mod = 998244353;
int main()
{
    int N, M;
    scanf("%d %d", &N, &M);
    
    int i, j;
    long long ans = 0, pow[3001], sum = 0, r_sum[3001] = {}, dp[3001] = {}, tmp[3001][3001];
    for (i = 1, pow[0] = 1; i <= N; i++) pow[i] = pow[i-1] * M % Mod;
    for (j = 2; j <= M; j++) {
        tmp[1][j] = 1;
        r_sum[j] = 1;
        dp[j] = 1;
    }
    for (i = 2, sum = M - 1, ans = pow[N-1]; i <= N; i++) {
        ans += (dp[i] + sum) % Mod * pow[N-i] % Mod;
        for (j = 2; j <= M; j++) {
            tmp[i][j] = sum - r_sum[j];
            if (tmp[i][j] < 0) tmp[i][j] += Mod;
            if (i + j - 1 <= N) dp[i+j-1] += tmp[i][j];
            r_sum[j] += tmp[i][j];
            if (i >= j) r_sum[j] -= tmp[i-j+1][j];
            if (r_sum[j] < 0) r_sum[j] += Mod;
            else if (r_sum[j] >= Mod) r_sum[j] -= Mod;
        }
        for (j = 2; j <= M; j++) sum += tmp[i][j];
        sum = (sum + Mod - dp[i] % Mod) % Mod;
    }
    printf("%lld\n", ans % Mod);
    fflush(stdout);
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0