結果

問題 No.2487 Multiple of M
ユーザー Carpenters-Cat
提出日時 2023-09-29 22:30:20
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 981 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-07-23 06:54:33
合計ジャッジ時間 3,370 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 RE * 2
other AC * 4 RE * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import *
N, M, K = map(int, input().split())
mod = 998244353
g = gcd(M, K)
if N == 1 :
	print(0)
	exit()
if (M - 1) % mod == 0 :
	ans = 0
	if N & 1 :
		ans = 1
	else :
		ans = mod - 1
	ans = (ans * (g - 1)) % mod
	print(ans)
else :
	X = pow(M - 1, N - 1, mod)
	iv = pow(M - 1, -1, mod)
	n = N - 1
	mat = [[mod - iv, (g * iv) % mod], 0, 1]
	ret = [[1, 0], [0, 1]]
	while n > 0 :
		if n & 1 :
			ret2 = [[0, 0], [0, 0]]
			for i in range(2) :
				for j in range(2) :
					for k in range(2) :
						ret2[i][j] += ret[i][k] * mat[k][j]
						ret2[i][j] %= mod
			for i in range(2) :
				for j in range(2) :
					ret[i][j] = ret2[i][j]
		n >>= 1
		ret2 = [[0, 0], [0, 0]]
		for i in range(2) :
			for j in range(2) :
				for k in range(2) :
					ret2[i][j] += mat[i][k] * mat[k][j]
					ret2[i][j] %= mod
		for i in range(2) :
			for j in range(2) :
				mat[i][j] = ret2[i][j]
	b = (g - 1) * iv
	b %= mod
	bn = b * ret[0][0] + ret[0][1]
	bn %= mod
	print((X + mod - bn) % mod)
0