結果

問題 No.2255 Determinant Sum
ユーザー sotanishy
提出日時 2023-03-24 22:12:40
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,601 bytes
コンパイル時間 316 ms
コンパイル使用メモリ 82,316 KB
実行使用メモリ 77,448 KB
最終ジャッジ日時 2024-09-18 17:11:06
合計ジャッジ時間 3,096 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 13 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

def rref(A):
    if not A:
        return
    n, m = len(A), len(A[0])
    pivot = 0
    for j in range(m):
        i = pivot
        while i < n and A[i][j] == 0:
            i += 1
        if i == n:
            continue
        if i != pivot:
            A[i], A[pivot] = A[pivot], A[i]

        p = A[pivot][j]
        for l in range(j, m):
            A[pivot][l] //= p

        for k in range(n):
            if k == pivot:
                continue
            v = A[k][j]
            for l in range(j, m):
                A[k][l] -= A[pivot][l] * v

        pivot += 1

T = int(input())
for _ in range(T):
    N, P = map(int, input().split())
    A = [list(map(int, input().split())) for _ in range(N)]
    if P >= 3:
        print(0)
    else:
        row = [0] * N
        col = [0] * N
        ok = True
        for i in range(N):
            for j in range(N):
                if A[i][j] == -1:
                    if row[i] or col[j]:
                        ok = False
                        break
                    row[i] = 1
                    col[j] = 1
            if not ok:
                break

        if not ok:
            print(0)
        else:
            M = N - sum(row)
            rs = [i for i in range(N) if not row[i]]
            cs = [j for j in range(N) if not col[j]]
            rem = [[0] * M for _ in range(M)]
            for ni, i in enumerate(rs):
                for nj, j in enumerate(cs):
                    rem[ni][nj] = A[i][j]
            rref(rem)
            print(1 if not rem or sum(rem[-1]) > 0 else 0)
0