結果
問題 |
No.3226 2×2行列累乗
|
ユーザー |
|
提出日時 | 2025-08-13 15:37:08 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,514 bytes |
コンパイル時間 | 2,359 ms |
コンパイル使用メモリ | 203,752 KB |
実行使用メモリ | 7,716 KB |
最終ジャッジ日時 | 2025-08-13 15:37:12 |
合計ジャッジ時間 | 3,813 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; typedef long double ld; typedef pair<int, int> Pii; typedef pair<int, ll> Pil; typedef pair<ll, ll> Pll; typedef pair<ll, int> Pli; typedef vector<vector<ll>> Mat; #define fi first #define se second const ll MOD = 1e9 + 7; const ll MOD2 = 998244353; const ll MOD3 = 1812447359; const ll INF = 1ll << 62; const double PI = 2 * asin(1); void yes() { printf("yes\n"); } void no() { printf("no\n"); } void Yes() { printf("Yes\n"); } void No() { printf("No\n"); } void YES() { printf("YES\n"); } void NO() { printf("NO\n"); } Mat mat_calc(Mat a, Mat b, ll mod) { Mat ans = {{0, 0}, {0, 0}}; for (int i = 0; i < 2; i++) { for (int j = 0; j < 2; j++) { for (int k = 0; k < 2; k++) { ans[i][j] += a[i][k] * b[k][j] % mod; ans[i][j] %= mod; if (ans[i][j] < 0) { ans[i][j] += mod; } } } } return ans; } Mat mat_pow(Mat a, ll n, ll mod) { if (n == 0) { return {{1, 0}, {0, 1}}; } else if (n % 2 == 0) { return mat_pow(mat_calc(a, a, mod), n / 2, mod); } else { return mat_calc(a, mat_pow(a, n - 1, mod), mod); } } ll a, b, c, d, s, t, n, k; int main() { cin >> a >> b; cin >> c >> d; cin >> s >> t; cin >> n >> k; Mat p = mat_pow({{a, b}, {c, d}}, n, k); ll r = (p[0][0] * s + p[0][1] * t) % k; if (r < 0) { r += k; } ll u = (p[1][0] * s + p[1][1] * t) % k; if (u < 0) { u += k; } cout << r << " " << u << endl; return 0; }