結果

問題 No.1887 K Consecutive Ks (Easy)
ユーザー 👑 ygussanyygussany
提出日時 2022-03-07 15:45:13
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 899 bytes
コンパイル時間 2,069 ms
コンパイル使用メモリ 29,884 KB
実行使用メモリ 72,192 KB
最終ジャッジ日時 2023-09-29 16:35:10
合計ジャッジ時間 3,318 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 96 ms
72,192 KB
testcase_03 AC 0 ms
4,376 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 0 ms
4,380 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 1 ms
4,376 KB
testcase_10 AC 15 ms
69,968 KB
testcase_11 AC 15 ms
27,064 KB
testcase_12 AC 19 ms
67,396 KB
testcase_13 AC 89 ms
70,556 KB
testcase_14 AC 94 ms
72,060 KB
testcase_15 AC 93 ms
71,644 KB
testcase_16 AC 80 ms
71,780 KB
testcase_17 AC 49 ms
51,380 KB
testcase_18 AC 93 ms
72,064 KB
権限があれば一括ダウンロードができます

ソースコード

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;
}
0