結果
問題 |
No.3225 2×2行列相似判定 〜easy〜
|
ユーザー |
![]() |
提出日時 | 2025-08-08 21:34:16 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 3,994 bytes |
コンパイル時間 | 3,991 ms |
コンパイル使用メモリ | 260,104 KB |
実行使用メモリ | 7,716 KB |
最終ジャッジ日時 | 2025-08-08 21:34:27 |
合計ジャッジ時間 | 7,980 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 TLE * 1 -- * 1 |
other | -- * 33 |
ソースコード
#include <bits/stdc++.h> #include <cassert> #include <atcoder/all> using namespace std; using namespace atcoder; using ll = long long; using mint = modint; //using mint = modint998244353; //using mint = modint1000000007; #define rep(i, n) for (int i = 0; i < (int)(n); i++) #define repu(i, s, t) for (int i = (int)(s); i < (int)(t); i++) #define repd(i, s, t) for (int i = (int)(s)-1; i >= (int)(t); i--) #define all(v) v.begin(), v.end() void _u() { cerr << endl; } template <class H, class... T> void _u(H&& h, T&&... t) { cerr << h << ", "; _u(move(t)...); } #define U(...) { cerr << #__VA_ARGS__ << ": "; _u(__VA_ARGS__); } template<typename T> bool chmax(T &a, const T b) { if(a >= b) return false; a = b; return true; } template<typename T> bool chmin(T &a, const T b) { if(a <= b) return false; a = b; return true; } template<typename T> istream& operator>>(istream &in, vector<T> &a) { for(T &x: a) in >> x; return in; } template<typename T> ostream& operator<<(ostream &out, const vector<T> &a) { for(const T &x: a) out << x << ' '; return out; } const int di[] = {1, 0, -1, 0, 1, 1, -1, -1, 0}; const int dj[] = {0, 1, 0, -1, -1, 1, 1, -1, 0}; template <typename T=double> struct Matrix { vector<vector<T>> mat; int m, n; Matrix(initializer_list<initializer_list<T>> init) { mat.resize(0); for(const auto &row : init) { mat.emplace_back(row); } m = mat.size(); n = mat.empty() ? 0 : mat[0].size(); } Matrix(vector<vector<T>> &mat_): mat(mat_), m(int(mat_.size())), n(mat.empty() ? 0 : int(mat_[0].size())) {} Matrix(int x, int y):mat(vector(x, vector<T>(y))), m(x), n(y) {} Matrix(int x): m(x), n(x) { mat = vector(x, vector<T> (x)); rep(i, x) mat[i][i] = 1; } vector<T> &operator[](int x) { return mat[x]; } Matrix<T> &operator+=(const Matrix<T> &other) { assert(m == other.m && n == other.n); rep(i, m) rep(j, n) *this[i][j] += other[i][j]; return *this; } Matrix<T> operator+(const Matrix<T> &other) { return Matrix<T>(*this) += other; } Matrix<T> &operator*=(const Matrix<T> &other) { return *this = *this * other; } Matrix<T> operator*(const Matrix<T> &other) { assert(n == other.m); Matrix<T> result(m, other.n); rep(i, m) rep(j, other.n) rep(k, n) result.mat[i][j] += mat[i][k] * other.mat[k][j]; return result; } Matrix<T> pow(ll x) { assert(m == n); if(x == 0) return Matrix<T>(m); Matrix<T> result = pow(x / 2); result = result * result; return (x % 2 == 0 ? result : result * *this); } T dat() { assert(m == n); T result = 1; rep(i, m) { int pivot = -1; repu(j, i, m) if(mat[j][i] != 0) { pivot = j; break; } if(pivot == -1) return 0; if(pivot != i) { swap(mat[i], mat[pivot]); result *= -1; } result *= mat[i][i]; T inv = 1 / mat[i][i]; repu(j, i + 1, m) { T factor = mat[j][i] * inv; repu(k, i, n) mat[j][k] -= mat[i][k] * factor; } } return result; } }; int main() { cin.tie(0)->sync_with_stdio(0); const int p = 67; int a11, a12, a21, a22, b11, b12, b21, b22; cin >> a11 >> a12 >> a21 >> a22 >> b11 >> b12 >> b21 >> b22; mint::set_mod(p); Matrix<mint> A({{a11, a12}, {a21, a22}}), B({{b11, b12}, {b21, b22}}); rep(p11, p) rep(p12, p) rep(p21, p) rep(p22, p) { if(mint(p11) * p22 == mint(p12) * p21) continue; Matrix<mint> P({{p11, p12}, {p21, p22}}); auto res1 = P*A; auto res2 = B*P; bool ok = true; rep(i, 2) rep(j, 2) if(res1[i][j] != res2[i][j]) ok = false; if(ok) { cout << "Yes" << endl; return 0; } } cout << "No" << endl; return 0; }