結果

問題 No.2405 Minimal Matrix Decomposition
ユーザー gr1msl3y
提出日時 2023-08-17 17:41:22
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,676 bytes
コンパイル時間 557 ms
コンパイル使用メモリ 82,024 KB
実行使用メモリ 79,472 KB
最終ジャッジ日時 2024-11-26 16:07:00
合計ジャッジ時間 12,463 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 42 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

def prod(A, B, MOD):
    n = len(A)
    m = len(B[0])
    res = [[0]*m for _ in range(n)]
    for i in range(n):
        for j in range(m):
            for k in range(len(B)):
                res[i][j] += A[i][k]*B[k][j] % MOD
                res[i][j] %= MOD
    return res


P = int(input())
N, M = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(N)]
if A == [[0]*M for _ in range(N)]:
    if min(N, M) == 1:
        print(1)
        print(N, M)
        for _ in range(N):
            print(*[0]*M)
    else:
        print(2)
        print(N, 1)
        for _ in range(N):
            print(0)
        print(1, M)
        print(*[0]*M)
    exit()

B = [a[:] for a in A]
for r in range(N):
    for i in range(r, N):
        flag = 0
        for p in range(M):
            if B[i][p] != 0:
                flag = 1
                break
        if flag:
            break
    else:
        break
    B[r], B[i] = B[i], B[r]
    inv = pow(B[r][p], P-2, P)
    for j in range(M):
        B[r][j] = B[r][j]*inv % P
    for i in range(N):
        if i == r:
            continue
        v = B[i][p]
        for j in range(M):
            B[i][j] -= v*B[r][j] % P
            B[i][j] %= P

r = min(r, M)
if r*(N+M) >= N*M:
    print(1)
    print(N, M)
    for a in A:
        print(*a)
    exit()

plist = []
p = 0
for i in range(N):
    while p < M:
        if B[i][p] == 1:
            break
        p += 1
    else:
        break
    plist.append(p)
    p += 1

res0 = [[a[ind] for ind in plist] for a in A]
res1 = B[:r]
print(2)
print(N, r)
for res in res0:
    print(*res)
print(r, M)
for res in res1:
    print(*res)

# print(prod(res0,res1,P))
0