結果
問題 |
No.3226 2×2行列累乗
|
ユーザー |
![]() |
提出日時 | 2025-08-08 21:57:15 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 2,133 bytes |
コンパイル時間 | 5,198 ms |
コンパイル使用メモリ | 335,172 KB |
実行使用メモリ | 7,716 KB |
最終ジャッジ日時 | 2025-08-08 21:57:27 |
合計ジャッジ時間 | 6,189 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; using mint = atcoder::modint; //using mint = modint1000000007; //const int mod = 1000000007; //using mint = modint998244353; //const int mod = 998244353; //const int INF = 1e9; //const long long LINF = 1e18; #define rep(i, n) for (int i = 0; i < (n); ++i) #define rep2(i,l,r)for(int i=(l);i<(r);++i) #define rrep(i, n) for (int i = (n) - 1; i >= 0; --i) #define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i) #define all(x) (x).begin(),(x).end() #define allR(x) (x).rbegin(),(x).rend() #define P pair<int,int> template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; } template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; } #include <vector> template<typename T> std::vector<vector<T>> matrixMul(const std::vector<std::vector<T>>&A, const std::vector<std::vector<T>>&B) { std::vector<vector<T>> C(A.size(), std::vector<T>(B[0].size())); for (int i = 0; i < (int)A.size(); ++i) { for (int k = 0; k < (int)A[0].size(); ++k) { for (int j = 0; j < (int)B[0].size(); ++j) { C[i][j] += A[i][k] * B[k][j]; } } } return C; } template<typename T> std::vector<std::vector<T>> matrixPow(long long n, const std::vector<std::vector<T>> &mat) { int size = (int)mat.size(); vector<vector<mint>> mret(size, vector<mint>(size)); for (int i = 0; i < size; ++i) { mret[i][i] = 1; } auto cmat = mat; while (n > 0) { if (1 & n) mret = matrixMul(cmat, mret); cmat = matrixMul(cmat, cmat); n >>= 1; } return mret; } int main() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); int a, b, c, d; cin >> a >> b >> c >> d; int s, t; cin >> s >> t; long long n, k; cin >> n >> k; mint::set_mod(k); vector mat(2, vector<mint>(2)); mat[0][0] = a; mat[0][1] = b; mat[1][0] = c; mat[1][1] = d; mat = matrixPow(n, mat); vector mat2(2, vector<mint>(1)); mat2[0][0] = s; mat2[1][0] = t; mat = matrixMul(mat, mat2); cout << mat[0][0].val() << " " << mat[1][0].val() << endl; return 0; }