結果
問題 |
No.3226 2×2行列累乗
|
ユーザー |
![]() |
提出日時 | 2025-08-08 21:52:53 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,790 bytes |
コンパイル時間 | 4,458 ms |
コンパイル使用メモリ | 257,868 KB |
実行使用メモリ | 7,716 KB |
最終ジャッジ日時 | 2025-08-08 21:52:59 |
合計ジャッジ時間 | 5,759 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using isize = size_t; using i32 = int; using u32 = unsigned int; using i64 = long long; using u64 = unsigned long long; using i128 = __int128_t; using u128 = __uint128_t; using f64 = long double; using p2 = pair<i64, i64>; using el = tuple<i64, i64, i64>; using mint = atcoder::modint998244353; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); } i64 pow(i64 x, i64 n) { i64 res = 1; i64 t = x; while (n > 0) { if (n & 1) { res = res * t; } t = t * t; n >>= 1; } return res; } i64 pow(i64 x, i64 n, i64 m) { i64 res = 1; i64 t = x % m; while (n > 0) { if (n & 1) { res = res * t % m; } t = t * t % m; n >>= 1; } return res; } using mat = vector<vector<i64>>; using v4 = vector<i64>; mat mul(mat a, mat b, i64 K) { assert(a[0].size() == b.size()); mat res(a.size(), v4(b[0].size(), 0)); for (i64 i = 0; i < a.size(); i++) { for (i64 j = 0; j < b[0].size(); j++) { for (i64 k = 0; k < a[0].size(); k++) { res[i][j] += a[i][k] * b[k][j]; res[i][j] = (res[i][j] % K + K) % K; } } } return res; } mat pow(mat a, i64 n, i64 k) { mat t = a; assert(a.size() == a[0].size()); mat res(a.size(), v4(a.size(), 0)); for (i64 i = 0; i < a.size(); i++) res[i][i] = 1; while (n > 0) { if (n & 1) res = mul(res, t, k); t = mul(t, t, k); n >>= 1; } return res; } void _main() { mat m(2, v4(2, 0)); for (i64 i = 0; i < 2; i++) { for (i64 j = 0; j < 2; j++) { cin >> m[i][j]; } } i64 s, t, n, k; cin >> s >> t >> n >> k; m = pow(m, n, k); mat ppp = {{s}, {t}}; mat res = mul(m, ppp, k); cout << res[0][0] << " " << res[1][0] << "\n"; }