結果

問題 No.2405 Minimal Matrix Decomposition
コンテスト
ユーザー 獅子座じゃない人
提出日時 2023-07-29 12:43:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 182 ms / 2,000 ms
コード長 2,607 bytes
コンパイル時間 2,185 ms
コンパイル使用メモリ 176,968 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-08 04:53:22
合計ジャッジ時間 8,341 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#include <atcoder/modint>
using namespace atcoder;
using mint=modint;

int main(void)
{
    int p;
    cin >> p;
    int n,m;
    cin >> n >> m;
    mint::set_mod(p);
    vector<vector<mint>> a(n,vector<mint>(m));
    for(int i=0;i<n;++i){
        for(int j=0;j<m;++j){
            int atmp;
            cin >> atmp;
            a[i][j]=atmp;
        }
    }
    vector<vector<mint>> a_copy=a;
    int rank=0;
    for(int i=0;i<m;++i){
        if(rank==n){
            break;
        }
        if(a_copy[rank][i]==0){
            bool not_pivot=true;
            for(int j=rank+1;j<n;++j){
                if(a_copy[j][i]!=0){
                    not_pivot=false;
                    for(int k=i;k<m;++k){
                        swap(a_copy[rank][k],a_copy[j][k]);
                    }
                    break;
                }
            }
            if(not_pivot){
                continue;
            }
        }
        for(int j=m-1;j>=i;--j){
            a_copy[rank][j]/=a_copy[rank][i];
        }
        for(int j=0;j<n;++j){
            if(j==rank){
                continue;
            }
            for(int k=m-1;k>=i;--k){
                a_copy[j][k]-=a_copy[rank][k]*a_copy[j][i];
            }
        }
        ++rank;
    }
    if(rank==0){
        ++rank;
    }
    if(n*m<=n*rank+rank*m){
        cout << 1 << endl;
        cout << n << " " << m << endl;
        for(int i=0;i<n;++i){
            for(int j=0;j<m;++j){
                cout << a[i][j].val() << " ";
            }
            cout << endl;
        }
    } else {
        vector<vector<mint>> c(n,vector<mint>(rank));
        int pivot=0;
        for(int i=0;i<m;++i){
            if(pivot==n){
                break;
            }
            if(a_copy[pivot][i]==0){
                continue;
            }
            for(int j=0;j<n;++j){
                c[j][pivot]=a[j][i];
            }
            ++pivot;
        }
        vector<vector<mint>> f(rank,vector<mint>(m));
        for(int i=0;i<rank;++i){
            for(int j=0;j<m;++j){
                f[i][j]=a_copy[i][j];
            }
        }
        cout << 2 << endl;
        cout << n << " " << rank << endl;
        for(int i=0;i<n;++i){
            for(int j=0;j<rank;++j){
                cout << c[i][j].val() << " ";
            }
            cout << endl;
        }
        cout << rank << " " << m << endl;
        for(int i=0;i<rank;++i){
            for(int j=0;j<m;++j){
                cout << f[i][j].val() << " ";
            }
            cout << endl;
        }
    }
    return 0;
}
0