結果

問題 No.2825 Sum of Scores of Sets of Specified Sections
ユーザー 🦠みどりむし🦠みどりむし
提出日時 2024-01-22 16:31:48
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
実行時間 -
コード長 1,075 bytes
コンパイル時間 199 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 55,656 KB
最終ジャッジ日時 2024-07-26 20:00:57
合計ジャッジ時間 20,268 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 RE -
権限があれば一括ダウンロードができます

ソースコード

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