結果

問題 No.2444 一次変換と体積
ユーザー Nikkuniku029Nikkuniku029
提出日時 2023-08-25 23:19:31
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,080 bytes
コンパイル時間 211 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 45,684 KB
最終ジャッジ日時 2024-06-06 18:09:56
合計ジャッジ時間 12,225 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 464 ms
45,296 KB
testcase_01 AC 462 ms
45,156 KB
testcase_02 AC 483 ms
45,296 KB
testcase_03 WA -
testcase_04 RE -
testcase_05 RE -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 476 ms
45,032 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 AC 463 ms
45,032 KB
testcase_14 RE -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 457 ms
45,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np


def modinv(a: int, m: int) -> int:
    '''
    モジュラ逆元
    ax mod m =1の解x=a^(-1)を返す

    Parameters
    ----------
    a:int
    m:int
    '''
    x, y, u, v = 1, 0, 0, 1
    M = m
    while m > 0:
        k = a//m
        x -= k*u
        y -= k*v
        x, u = u, x
        y, v = v, y
        a, m = m, a % m
    assert a == 1, "a and m aren't relatively prime numbers"
    if x < 0:
        x += M
    return x


def mat_mul(a, b):
    """
    a: 行列(2次元配列)I*J
    b: 行列(2次元配列)J*K
    """
    I, J, K = len(a), len(b[0]), len(b)
    c = [[0] * J for _ in range(I)]
    for i in range(I):
        for j in range(J):
            for k in range(K):
                c[i][j] += a[i][k] * b[k][j]
    return c


N, B = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(3)]
mat = np.matrix(A)
rank = np.linalg.matrix_rank(mat)
if 0 < rank < 3:
    exit(print(0))
elif rank == 0:
    exit(print('infty'))
det = int(np.linalg.det(mat))
detn = pow(det, N, B)
ans = modinv(detn, B)
print(ans)
0