結果

問題 No.2487 Multiple of M
ユーザー Carpenters-Cat
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import gcd
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 - 2
    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
    bn = (bn * pow(M - 1, N - 1, mod)) % mod
    # print(bn)
    # print(X)
    print((X + mod - bn) % mod)
0