結果

問題 No.2405 Minimal Matrix Decomposition
ユーザー 獅子座じゃない人
提出日時 2023-07-28 14:15:55
言語 PyPy3
(7.3.15)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,464 bytes
コンパイル時間 549 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 80,504 KB
最終ジャッジ日時 2024-10-08 04:53:59
合計ジャッジ時間 10,959 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 43 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

import copy

p=int(input())
n,m=map(int,input().split())
a=[list(map(int,input().split())) for i in range(n)]
a_copy=copy.deepcopy(a)
rank=0
for i in range(m):
    if rank==n:
        break
    if a_copy[rank][i]==0:
        not_pivot=True
        for j in range(rank+1,n):
            if a_copy[j][i]!=0:
                not_pivot=False
                for k in range(i,m):
                    a_copy[rank][k],a_copy[j][k]=a_copy[j][k],a_copy[rank][k]
                break
        if not_pivot:
            continue
    for j in reversed(range(i,m)):
        a_copy[rank][j]*=pow(a_copy[rank][i],p-2,p)
        a_copy[rank][j]%=p
    for j in range(0,n):
        if j==rank:
            continue
        for k in reversed(range(i,m)):
            a_copy[j][k]-=a_copy[rank][k]*a_copy[j][i]
            a_copy[j][k]%=p
    rank+=1
if n*m<=n*rank+rank*m:
    print(1)
    print(n,m)
    for i in range(n):
        print(*a[i])
else:
    c=[[0 for j in range(rank)] for i in range(n)]
    pivot=0
    for i in range(m):
        if pivot==n:
            break
        if a_copy[pivot][i]==0:
            continue
        for j in range(n):
            c[j][pivot]=a[j][i]
        pivot+=1
    f=[[0 for j in range(m)] for i in range(rank)]
    for i in range(rank):
        for j in range(m):
            f[i][j]=a_copy[i][j]
    print(2)
    print(n,rank)
    for i in range(n):
        print(*c[i])
    print(rank,m)
    for i in range(rank):
        print(*f[i])
0