結果

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