#899164 (C++23) No.2405 Minimal Matrix Decomposition

提出ソース
結果

問題	No.2405 Minimal Matrix Decomposition
ユーザー	👑 rin204
提出日時	2023-08-05 19:23:52
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	915 ms / 2,000 ms
コード長	20,673 bytes
コンパイル時間	4,121 ms
コンパイル使用メモリ	265,844 KB
実行使用メモリ	5,888 KB
最終ジャッジ日時	2024-10-15 16:32:15
合計ジャッジ時間	12,021 ms
ジャッジサーバーID （参考情報）	judge1 / judge4
このコードへのチャレンジ
（要ログイン）
テストケース

テストケース表示
入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,248 KB
testcase_03	AC	1 ms 5,248 KB
testcase_04	AC	19 ms 5,888 KB
testcase_05	AC	915 ms 5,248 KB
testcase_06	AC	2 ms 5,248 KB
testcase_07	AC	2 ms 5,248 KB
testcase_08	AC	2 ms 5,248 KB
testcase_09	AC	2 ms 5,248 KB
testcase_10	AC	246 ms 5,248 KB
testcase_11	AC	520 ms 5,760 KB
testcase_12	AC	4 ms 5,248 KB
testcase_13	AC	73 ms 5,248 KB
testcase_14	AC	60 ms 5,248 KB
testcase_15	AC	32 ms 5,248 KB
testcase_16	AC	16 ms 5,248 KB
testcase_17	AC	183 ms 5,248 KB
testcase_18	AC	72 ms 5,248 KB
testcase_19	AC	66 ms 5,248 KB
testcase_20	AC	3 ms 5,248 KB
testcase_21	AC	23 ms 5,248 KB
testcase_22	AC	379 ms 5,248 KB
testcase_23	AC	3 ms 5,248 KB
testcase_24	AC	2 ms 5,248 KB
testcase_25	AC	6 ms 5,248 KB
testcase_26	AC	160 ms 5,248 KB
testcase_27	AC	67 ms 5,248 KB
testcase_28	AC	158 ms 5,248 KB
testcase_29	AC	24 ms 5,248 KB
testcase_30	AC	188 ms 5,248 KB
testcase_31	AC	62 ms 5,248 KB
testcase_32	AC	7 ms 5,248 KB
testcase_33	AC	252 ms 5,248 KB
testcase_34	AC	32 ms 5,248 KB
testcase_35	AC	2 ms 5,248 KB
testcase_36	AC	62 ms 5,248 KB
testcase_37	AC	124 ms 5,248 KB
testcase_38	AC	90 ms 5,248 KB
testcase_39	AC	187 ms 5,248 KB
testcase_40	AC	85 ms 5,248 KB
testcase_41	AC	8 ms 5,248 KB
testcase_42	AC	319 ms 5,248 KB
testcase_43	AC	49 ms 5,248 KB
testcase_44	AC	35 ms 5,248 KB
testcase_45	AC	10 ms 5,248 KB
testcase_46	AC	31 ms 5,248 KB
testcase_47	AC	46 ms 5,248 KB
testcase_48	AC	20 ms 5,248 KB
testcase_49	AC	4 ms 5,248 KB
権限があれば一括ダウンロードができます
ソースコード

raw source code
// start A.cpp
// #pragma GCC target("avx2")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;

using ll  = long long;
using ull = unsigned long long;
template <class T>
using pq = priority_queue<T>;
template <class T>
using qp = priority_queue<T, vector<T>, greater<T>>;
#define vec(T, A, ...) vector<T> A(__VA_ARGS__);
#define vvec(T, A, h, ...) vector<vector<T>> A(h, vector<T>(__VA_ARGS__));
#define vvvec(T, A, h1, h2, ...) vector<vector<vector<T>>> A(h1, vector<vector<T>>(h2, vector<T>(__VA_ARGS__)));

#ifndef RIN__LOCAL
#define endl "\n"
#endif
#define spa ' '
#define len(A) A.size()
#define all(A) begin(A), end(A)

#define fori1(a) for (ll _ = 0; _ < (a); _++)
#define fori2(i, a) for (ll i = 0; i < (a); i++)
#define fori3(i, a, b) for (ll i = (a); i < (b); i++)
#define fori4(i, a, b, c) for (ll i = (a); ((c) > 0 || i > (b)) && ((c) < 0 || i < (b)); i += (c))
#define overload4(a, b, c, d, e, ...) e
#define fori(...) overload4(__VA_ARGS__, fori4, fori3, fori2, fori1)(__VA_ARGS__)

vector<char> stoc(string &S) {
    int n = S.size();
    vector<char> ret(n);
    for (int i = 0; i < n; i++) ret[i] = S[i];
    return ret;
}

#define INT(...)                                                                                                                                                                                       \
    int __VA_ARGS__;                                                                                                                                                                                   \
    inp(__VA_ARGS__);
#define LL(...)                                                                                                                                                                                        \
    ll __VA_ARGS__;                                                                                                                                                                                    \
    inp(__VA_ARGS__);
#define STRING(...)                                                                                                                                                                                    \
    string __VA_ARGS__;                                                                                                                                                                                \
    inp(__VA_ARGS__);
#define CHAR(...)                                                                                                                                                                                      \
    char __VA_ARGS__;                                                                                                                                                                                  \
    inp(__VA_ARGS__);
#define VEC(T, A, n)                                                                                                                                                                                   \
    vector<T> A(n);                                                                                                                                                                                    \
    inp(A);
#define VVEC(T, A, n, m)                                                                                                                                                                               \
    vector<vector<T>> A(n, vector<T>(m));                                                                                                                                                              \
    inp(A);

const ll MOD1 = 1000000007;
const ll MOD9 = 998244353;

template <class T>
auto min(const T &a) {
    return *min_element(all(a));
}
template <class T>
auto max(const T &a) {
    return *max_element(all(a));
}
template <class T, class S>
auto clamp(T &a, const S &l, const S &r) {
    return (a > r ? r : a < l ? l : a);
}
template <class T, class S>
inline bool chmax(T &a, const S &b) {
    return (a < b ? a = b, 1 : 0);
}
template <class T, class S>
inline bool chmin(T &a, const S &b) {
    return (a > b ? a = b, 1 : 0);
}
template <class T, class S>
inline bool chclamp(T &a, const S &l, const S &r) {
    auto b = clamp(a, l, r);
    return (a != b ? a = b, 1 : 0);
}

void FLUSH() {
    cout << flush;
}
void print() {
    cout << endl;
}
template <class Head, class... Tail>
void print(Head &&head, Tail &&...tail) {
    cout << head;
    if (sizeof...(Tail)) cout << spa;
    print(forward<Tail>(tail)...);
}
template <typename T>
void print(vector<T> &A) {
    int n = A.size();
    for (int i = 0; i < n; i++) {
        cout << A[i];
        if (i != n - 1) cout << ' ';
    }
    cout << endl;
}
template <typename T>
void print(vector<vector<T>> &A) {
    for (auto &row : A) print(row);
}
template <typename T, typename S>
void print(pair<T, S> &A) {
    cout << A.first << spa << A.second << endl;
}
template <typename T, typename S>
void print(vector<pair<T, S>> &A) {
    for (auto &row : A) print(row);
}
template <typename T, typename S>
void prisep(vector<T> &A, S sep) {
    int n = A.size();
    for (int i = 0; i < n; i++) {
        cout << A[i];
        if (i != n - 1) cout << sep;
    }
    cout << endl;
}
template <typename T, typename S>
void priend(T A, S end) {
    cout << A << end;
}
template <typename T>
void priend(T A) {
    priend(A, spa);
}
template <typename T, typename S>
bool printif(bool f, T A, S B) {
    if (f)
        print(A);
    else
        print(B);
    return f;
}
template <class... T>
void inp(T &...a) {
    (cin >> ... >> a);
}
template <typename T>
void inp(vector<T> &A) {
    for (auto &a : A) cin >> a;
}
template <typename T>
void inp(vector<vector<T>> &A) {
    for (auto &row : A) inp(row);
}
template <typename T, typename S>
void inp(pair<T, S> &A) {
    inp(A.first, A.second);
}
template <typename T, typename S>
void inp(vector<pair<T, S>> &A) {
    for (auto &row : A) inp(row.first, row.second);
}

template <typename T>
T sum(vector<T> &A) {
    T tot = 0;
    for (auto a : A) tot += a;
    return tot;
}

template <typename T>
vector<T> compression(vector<T> X) {
    sort(all(X));
    X.erase(unique(all(X)), X.end());
    return X;
}

vector<vector<int>> read_edges(int n, int m, bool direct = false, int indexed = 1) {
    vector<vector<int>> edges(n, vector<int>());
    for (int i = 0; i < m; i++) {
        INT(u, v);
        u -= indexed;
        v -= indexed;
        edges[u].push_back(v);
        if (!direct) edges[v].push_back(u);
    }
    return edges;
}
vector<vector<int>> read_tree(int n, int indexed = 1) {
    return read_edges(n, n - 1, false, indexed);
}
template <typename T>
vector<vector<pair<int, T>>> read_wedges(int n, int m, bool direct = false, int indexed = 1) {
    vector<vector<pair<int, T>>> edges(n, vector<pair<int, T>>());
    for (int i = 0; i < m; i++) {
        INT(u, v);
        T w;
        inp(w);
        u -= indexed;
        v -= indexed;
        edges[u].push_back({v, w});
        if (!direct) edges[v].push_back({u, w});
    }
    return edges;
}
template <typename T>
vector<vector<pair<int, T>>> read_wtree(int n, int indexed = 1) {
    return read_wedges<T>(n, n - 1, false, indexed);
}

inline bool yes(bool f = true) {
    cout << (f ? "yes" : "no") << endl;
    return f;
}
inline bool Yes(bool f = true) {
    cout << (f ? "Yes" : "No") << endl;
    return f;
}
inline bool YES(bool f = true) {
    cout << (f ? "YES" : "NO") << endl;
    return f;
}

inline bool no(bool f = true) {
    cout << (!f ? "yes" : "no") << endl;
    return f;
}
inline bool No(bool f = true) {
    cout << (!f ? "Yes" : "No") << endl;
    return f;
}
inline bool NO(bool f = true) {
    cout << (!f ? "YES" : "NO") << endl;
    return f;
}

// start other/Modint.hpp

template <int MOD>
struct Modint {
    int x;
    Modint() : x(0) {}
    Modint(int64_t y) {
        if (y >= 0)
            x = y % MOD;
        else
            x = (y % MOD + MOD) % MOD;
    }

    Modint &operator+=(const Modint &p) {
        x += p.x;
        if (x >= MOD) x -= MOD;
        return *this;
    }

    Modint &operator-=(const Modint &p) {
        x -= p.x;
        if (x < 0) x += MOD;
        return *this;
    }

    Modint &operator*=(const Modint &p) {
        x = int(1LL * x * p.x % MOD);
        return *this;
    }

    Modint &operator/=(const Modint &p) {
        *this *= p.inverse();
        return *this;
    }

    Modint &operator%=(const Modint &p) {
        assert(p.x == 0);
        return *this;
    }

    Modint operator-() const {
        return Modint(-x);
    }

    Modint &operator++() {
        x++;
        if (x == MOD) x = 0;
        return *this;
    }

    Modint &operator--() {
        if (x == 0) x = MOD;
        x--;
        return *this;
    }

    Modint operator++(int) {
        Modint result = *this;
        ++*this;
        return result;
    }

    Modint operator--(int) {
        Modint result = *this;
        --*this;
        return result;
    }

    friend Modint operator+(const Modint &lhs, const Modint &rhs) {
        return Modint(lhs) += rhs;
    }

    friend Modint operator-(const Modint &lhs, const Modint &rhs) {
        return Modint(lhs) -= rhs;
    }

    friend Modint operator*(const Modint &lhs, const Modint &rhs) {
        return Modint(lhs) *= rhs;
    }

    friend Modint operator/(const Modint &lhs, const Modint &rhs) {
        return Modint(lhs) /= rhs;
    }

    friend Modint operator%(const Modint &lhs, const Modint &rhs) {
        assert(rhs.x == 0);
        return Modint(lhs);
    }

    bool operator==(const Modint &p) const {
        return x == p.x;
    }

    bool operator!=(const Modint &p) const {
        return x != p.x;
    }

    bool operator<(const Modint &rhs) const {
        return x < rhs.x;
    }

    bool operator<=(const Modint &rhs) const {
        return x <= rhs.x;
    }

    bool operator>(const Modint &rhs) const {
        return x > rhs.x;
    }

    bool operator>=(const Modint &rhs) const {
        return x >= rhs.x;
    }

    Modint inverse() const {
        int a = x, b = MOD, u = 1, v = 0, t;
        while (b > 0) {
            t = a / b;
            a -= t * b;
            u -= t * v;
            swap(a, b);
            swap(u, v);
        }
        return Modint(u);
    }

    Modint pow(int64_t k) const {
        Modint ret(1);
        Modint y(x);
        while (k > 0) {
            if (k & 1) ret *= y;
            y *= y;
            k >>= 1;
        }
        return ret;
    }

    friend ostream &operator<<(ostream &os, const Modint &p) {
        return os << p.x;
    }

    friend istream &operator>>(istream &is, Modint &p) {
        int64_t y;
        is >> y;
        p = Modint<MOD>(y);
        return (is);
    }

    static int get_mod() {
        return MOD;
    }
};

struct Arbitrary_Modint {
    int x;
    static int MOD;

    static void set_mod(int mod) {
        MOD = mod;
    }

    Arbitrary_Modint() : x(0) {}
    Arbitrary_Modint(int64_t y) {
        if (y >= 0)
            x = y % MOD;
        else
            x = (y % MOD + MOD) % MOD;
    }

    Arbitrary_Modint &operator+=(const Arbitrary_Modint &p) {
        x += p.x;
        if (x >= MOD) x -= MOD;
        return *this;
    }

    Arbitrary_Modint &operator-=(const Arbitrary_Modint &p) {
        x -= p.x;
        if (x < 0) x += MOD;
        return *this;
    }

    Arbitrary_Modint &operator*=(const Arbitrary_Modint &p) {
        x = int(1LL * x * p.x % MOD);
        return *this;
    }

    Arbitrary_Modint &operator/=(const Arbitrary_Modint &p) {
        *this *= p.inverse();
        return *this;
    }

    Arbitrary_Modint &operator%=(const Arbitrary_Modint &p) {
        assert(p.x == 0);
        return *this;
    }

    Arbitrary_Modint operator-() const {
        return Arbitrary_Modint(-x);
    }

    Arbitrary_Modint &operator++() {
        x++;
        if (x == MOD) x = 0;
        return *this;
    }

    Arbitrary_Modint &operator--() {
        if (x == 0) x = MOD;
        x--;
        return *this;
    }

    Arbitrary_Modint operator++(int) {
        Arbitrary_Modint result = *this;
        ++*this;
        return result;
    }

    Arbitrary_Modint operator--(int) {
        Arbitrary_Modint result = *this;
        --*this;
        return result;
    }

    friend Arbitrary_Modint operator+(const Arbitrary_Modint &lhs, const Arbitrary_Modint &rhs) {
        return Arbitrary_Modint(lhs) += rhs;
    }

    friend Arbitrary_Modint operator-(const Arbitrary_Modint &lhs, const Arbitrary_Modint &rhs) {
        return Arbitrary_Modint(lhs) -= rhs;
    }

    friend Arbitrary_Modint operator*(const Arbitrary_Modint &lhs, const Arbitrary_Modint &rhs) {
        return Arbitrary_Modint(lhs) *= rhs;
    }

    friend Arbitrary_Modint operator/(const Arbitrary_Modint &lhs, const Arbitrary_Modint &rhs) {
        return Arbitrary_Modint(lhs) /= rhs;
    }

    friend Arbitrary_Modint operator%(const Arbitrary_Modint &lhs, const Arbitrary_Modint &rhs) {
        assert(rhs.x == 0);
        return Arbitrary_Modint(lhs);
    }

    bool operator==(const Arbitrary_Modint &p) const {
        return x == p.x;
    }

    bool operator!=(const Arbitrary_Modint &p) const {
        return x != p.x;
    }

    bool operator<(const Arbitrary_Modint &rhs) {
        return x < rhs.x;
    }

    bool operator<=(const Arbitrary_Modint &rhs) {
        return x <= rhs.x;
    }

    bool operator>(const Arbitrary_Modint &rhs) {
        return x > rhs.x;
    }

    bool operator>=(const Arbitrary_Modint &rhs) {
        return x >= rhs.x;
    }

    Arbitrary_Modint inverse() const {
        int a = x, b = MOD, u = 1, v = 0, t;
        while (b > 0) {
            t = a / b;
            a -= t * b;
            u -= t * v;
            swap(a, b);
            swap(u, v);
        }
        return Arbitrary_Modint(u);
    }

    Arbitrary_Modint pow(int64_t k) const {
        Arbitrary_Modint ret(1);
        Arbitrary_Modint y(x);
        while (k > 0) {
            if (k & 1) ret *= y;
            y *= y;
            k >>= 1;
        }
        return ret;
    }

    friend ostream &operator<<(ostream &os, const Arbitrary_Modint &p) {
        return os << p.x;
    }

    friend istream &operator>>(istream &is, Arbitrary_Modint &p) {
        int64_t y;
        is >> y;
        p = Arbitrary_Modint(y);
        return (is);
    }

    static int get_mod() {
        return MOD;
    }
};
int Arbitrary_Modint::MOD = 998244353;

using modint9 = Modint<998244353>;
using modint1 = Modint<1000000007>;
using modint  = Arbitrary_Modint;

// end other/Modint.hpp
// restart A.cpp
using mint = modint;
// start math/Matrix.hpp

template <typename type>
struct Matrix {
    int n, m;
    vector<vector<type>> A;
    Matrix() = default;
    Matrix(int n, int m) : A(n, vector<type>(m, 0)), n(n), m(m) {}
    Matrix(int n) : A(n, vector<type>(n, 0)), n(n), m(n) {}
    Matrix(vector<vector<type>> A) : A(A), n(A.size()), m(A[0].size()) {}

    inline const vector<type> &operator[](int k) const {
        return (A.at(k));
    }

    inline vector<type> &operator[](int k) {
        return (A.at(k));
    }

    Matrix T() {
        Matrix<type> B(m, n);
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++) {
                B.A[i][j] = A[j][i];
            }
        return B;
    }

    Matrix &operator+=(const Matrix &B) {
        assert(n == B.A.size());
        assert(m == B.A[0].size());
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++) {
                this->A[i][j] += B[i][j];
            }
        return *this;
    }

    Matrix &operator-=(const Matrix &B) {
        assert(n == B.A.size());
        assert(m == B.A[0].size());
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++) {
                this->A[i][j] -= B[i][j];
            }
        return *this;
    }

    Matrix &operator*=(const Matrix &B) {
        int k = B[0].size();
        assert(m == B.A.size());
        vector<vector<type>> C(n, vector<type>(k, 0));
        for (int i = 0; i < n; i++)
            for (int j = 0; j < k; j++) {
                for (int l = 0; l < m; l++) {
                    C[i][j] += this->A[i][l] * B[l][j];
                }
            }
        swap(this->A, C);
        return *this;
    }

    Matrix operator+(const Matrix &B) const {
        return (Matrix(*this) += B);
    }

    Matrix operator-(const Matrix &B) const {
        return (Matrix(*this) -= B);
    }

    Matrix operator*(const Matrix &B) const {
        return (Matrix(*this) *= B);
    }

    type det() {
        auto arr = A;
        assert(n == m);
        type ret = 1;
        for (int i = 0; i < n; i++) {
            if (arr[i][i] == 0) {
                bool ng = true;
                for (int j = i + 1; j < n; j++) {
                    if (arr[j][i] == 0) continue;
                    swap(arr[i], arr[j]);
                    ret *= -1;
                    ng = false;
                    break;
                }
                if (ng) return 0;
            }
            ret *= arr[i][i];
            type inv = type(1) / arr[i][i];
            for (int j = i; j < n; j++) arr[i][j] *= inv;
            for (int j = i + 1; j < n; j++) {
                type x = arr[j][i];
                for (int k = i; k < n; k++) {
                    arr[j][k] -= arr[i][k] * x;
                }
            }
        }
        return ret;
    }

    void I() {
        assert(n == m);
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (i == j)
                    A[i][j] = 1;
                else
                    A[i][j] = 0;
            }
        }
    }

    Matrix<type> inv() {
        assert(n == m);
        Matrix<type> ret(n);
        ret.I();
        auto &B  = ret.A;
        auto arr = A;
        for (int j = 0; j < n; j++) {
            int ii = -1;
            for (int i = j; i < n; i++) {
                if (arr[i][j] != 0) {
                    ii = i;
                    break;
                }
            }
            if (ii == -1) {
                return {};
            }
            swap(arr[j], arr[ii]);
            swap(B[j], B[ii]);
            ii       = j;
            type inv = type(1) / arr[ii][j];

            for (int jj = 0; jj < n; jj++) {
                B[ii][jj] *= inv;
                arr[ii][jj] *= inv;
            }

            for (int i = 0; i < n; i++) {
                if (i == ii) continue;
                type t = arr[i][j];
                for (int jj = 0; jj < n; jj++) {
                    arr[i][jj] -= arr[ii][jj] * t;
                    B[i][jj] -= B[ii][jj] * t;
                }
            }
        }
        return ret;
    }
};

template <typename T>
void inp(Matrix<T> &M) {
    for (auto &row : M.A) inp(row);
}

template <typename T>
void print(Matrix<T> &M) {
    for (auto &row : M.A) print(row);
}
// end math/Matrix.hpp
// restart A.cpp

void solve() {
    INT(P);
    mint::set_mod(P);
    INT(n, m);
    VVEC(mint, A, n, m);
    auto B = A;
    vec(int, used, n, -1);
    vec(int, lind, 0);
    int rank = 0;
    fori(j, m) {
        int ind = -1;
        fori(i, n) {
            if (used[i] == -1 && A[i][j] != 0) {
                ind = i;
                break;
            }
        }
        if (ind == -1) continue;
        used[ind] = rank++;
        lind.push_back(j);
        mint inv = mint(1) / A[ind][j];
        fori(j, m) A[ind][j] *= inv;
        fori(i, n) {
            if (i == ind) continue;
            mint t = A[i][j];
            fori(k, m) {
                A[i][k] -= t * A[ind][k];
            }
        }
    }
    if (rank == 0) {
        rank++;
        used[0] = 0;
        lind.push_back(0);
    }

    if (rank * (n + m) > n * m) {
        print(1);
        print(n, m);
        print(B);
        return;
    }

    Matrix<mint> L(n, rank);
    Matrix<mint> R(rank, m);

    fori(j, rank) {
        fori(i, n) {
            L[i][j] = B[i][lind[j]];
        }
    }

    fori(i, n) {
        if (used[i] == -1) continue;
        fori(j, m) {
            R[used[i]][j] = A[i][j];
        }
    }

    print(2);
    print(n, rank);
    print(L);
    print(rank, m);
    print(R);

    auto X = L * R;
    fori(i, n) fori(j, m) {
        assert(B[i][j] == X[i][j]);
    }
}

int main() {
    cin.tie(0)->sync_with_stdio(0);
    // cout << fixed << setprecision(12);
    int t;
    t = 1;
    // cin >> t;
    while (t--) solve();
    return 0;
}

// end A.cpp
yukicoder

結果

テストケース

ソースコード
