結果

問題 No.2825 Sum of Scores of Sets of Specified Sections
コンテスト
ユーザー みどりむし🦠
提出日時 2024-07-26 20:05:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,205 ms / 3,000 ms
コード長 1,041 bytes
コンパイル時間 189 ms
コンパイル使用メモリ 82,584 KB
実行使用メモリ 146,784 KB
最終ジャッジ日時 2024-07-26 20:05:46
合計ジャッジ時間 8,835 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353


def determinant(A):
    A = [a.copy() for a in A]

    n = len(A)
    res = 1

    for i, row in enumerate(A):
        if row[i] == 0:
            for j in range(i + 1, n):
                if A[j][i] != 0:
                    break
            else:
                return 0

            A[i], A[j] = A[j], A[i]
            row = A[i]
            res *= -1

        inv = pow(row[i], MOD - 2, MOD)

        for j in range(i + 1, n):
            t = A[j][i] * inv % MOD

            for k in range(i + 1, n):
                A[j][k] -= t * row[k]
                A[j][k] %= MOD

    for i in range(n):
        res *= A[i][i]
        res %= MOD

    return res


n, m = map(int, input().split())

A = [[*map(int, input().split())] for _ in range(n)]
B = [*zip(*(map(int, input().split()) for _ in range(n)))]

C = [[0] * n for i in range(n)]

for k in range(m):
    for i in range(n):
        for j in range(n):
            C[i][j] += A[i][k] * B[k][j]

for i in range(n):
    C[i][i] += 1

print((determinant(C) - 1) % MOD)
0