結果

問題 No.2825 Sum of Scores of Sets of Specified Sections
ユーザー 🦠みどりむし🦠みどりむし
提出日時 2024-01-22 16:32:19
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 1,075 bytes
コンパイル時間 108 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 55,912 KB
最終ジャッジ日時 2024-07-26 20:00:55
合計ジャッジ時間 18,465 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 1
other WA * 4 RE * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

import numpy as np

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):
            a_j = A[j]
            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 = [[*map(int, input().split())] for _ in range(m)]

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))
0