結果

問題 No.3226 2×2行列累乗
ユーザー u_kun
提出日時 2025-08-08 22:03:46
言語 PyPy3
(8.0.1 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 34 ms / 2,000 ms
+ 997µs
コード長 840 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 95 ms
コンパイル使用メモリ 64,392 KB
実行使用メモリ 53,708 KB
最終ジャッジ日時 2026-11-11 05:58:40
合計ジャッジ時間 2,886 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge5_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def dot(A, B):
    
    C = [[None for col in range(2)] for row in range(2)]
    
    for row in range(2):
        for col in range(2):
            
            # C[row][col]:= Aのrow行目とBのcol列目の内積
            tmp = 0
            
            for i in range(2):
                tmp += ((A[row][i]%K) * (B[i][col]%K))%K
                tmp %= K
            
            C[row][col] = tmp
    
    return C
    
M = [list(map(int, input().split())) for row in range(2)]
S, T = map(int, input().split())
N, K = map(int, input().split())

bin_N = bin(N)
bin_N = bin_N[2:]
bin_N = bin_N[::-1]

if bin_N[0] == "1":
    A = M
else:
    A = [[1, 0], [0, 1]]

for i in range(1, len(bin_N)):
    M = dot(M, M)
    
    if bin_N[i] == "1":
        A = dot(A, M)

R = A[0][0]*S + A[0][1]*T
U = A[1][0]*S + A[1][1]*T

print(R%K, U%K)
0