結果

問題 No.2444 一次変換と体積
コンテスト
ユーザー Kude
提出日時 2023-08-25 21:47:12
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 810 bytes
コンパイル時間 163 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 52,992 KB
最終ジャッジ日時 2024-12-24 08:26:44
合計ジャッジ時間 1,655 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, b = map(int, input().split())
a = [list(map(int, input().split())) for _ in range(3)]

def mul(X, Y):
    res = [[0] * 3 for _ in range(3)]
    for i in range(3):
        for j in range(3):
            for k in range(3):
                res[i][k] += X[i][j] * Y[j][k]
    for i in range(3):
        for j in range(3):
            res[i][j] %= b
    return res

def det(a):
    res = 0
    for j in range(3):
        v = 1
        for i in range(3):
            v *= a[i][(j+i) % 3]
        res += v
    for j in range(3):
        v = 1
        for i in range(3):
            v *= a[i][(j-i) % 3]
        res -= v
    return res

an = [[int(i == j) for j in range(3)] for i in range(3)]
for k in range(n.bit_length())[::-1]:
    an = mul(an, an)
    if n >> k & 1:
        an = mul(an, a)

print(det(an) % b)
0