結果
問題 | No.2487 Multiple of M |
ユーザー |
|
提出日時 | 2023-09-29 22:48:59 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,347 bytes |
コンパイル時間 | 97 ms |
コンパイル使用メモリ | 13,056 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2024-07-23 06:54:41 |
合計ジャッジ時間 | 3,305 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 WA * 14 |
ソースコード
from math import gcdN, M, K = map(int, input().split())mod = 998244353g = gcd(M, K)if N == 1 :print(0)exit()if (M - 1) % mod == 0 :ans = 0if N & 1 :ans = 1else :ans = mod - 1ans = (ans * (g - 1)) % modprint(ans)else :X = pow(M - 1, N - 1, mod)iv = pow(M - 1, -1, mod)n = N - 2mat = [[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] %= modfor i in range(2) :for j in range(2) :ret[i][j] = ret2[i][j]n >>= 1ret2 = [[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] %= modfor i in range(2) :for j in range(2) :mat[i][j] = ret2[i][j]b = (g - 1) * ivb %= modbn = b * ret[0][0] + ret[0][1]bn %= modbn = (bn * pow(M - 1, N - 1, mod)) % mod# print(bn)# print(X)print((X + mod - bn) % mod)