結果
問題 | No.3006 ベイカーの問題 |
ユーザー |
![]() |
提出日時 | 2025-01-18 16:28:14 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,786 bytes |
コンパイル時間 | 5,076 ms |
コンパイル使用メモリ | 256,048 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2025-01-18 16:28:21 |
合計ジャッジ時間 | 5,153 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/all>using namespace std;using namespace atcoder;#define rep(i, n) for(int i=0;i<(n);++i)#define rep1(i, n) for(int i=1;i<=(n);i++)#define ll long longusing mint = modint998244353;using P = pair<ll,ll>;using lb = long double;using T = tuple<ll, ll, ll>;#ifdef LOCAL# include <debug_print.hpp># define dbg(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__)#else# define dbg(...) (static_cast<void>(0))#endiftemplate<typename T>struct Matrix {int h, w;vector<vector<T>> d;Matrix() {}//初期化のときは Matrix<> a(h, w)Matrix(int h, int w, T val=0): h(h), w(w), d(h, vector<T>(w,val)) {}Matrix& unit() {assert(h == w);rep(i,h) d[i][i] = 1;return *this;}const vector<T>& operator[](int i) const { return d[i];}vector<T>& operator[](int i) { return d[i];}Matrix operator*(const Matrix& a) const {assert(w == a.h);Matrix r(h, a.w);rep(i,h)rep(k,w)rep(j,a.w) {r[i][j] += d[i][k]*a[k][j];}return r;}//Matcix<> newa = a.pow(k)と使うMatrix pow(long long t) const {assert(h == w);if (!t) return Matrix(h,h).unit();if (t == 1) return *this;Matrix r = pow(t>>1);r = r*r;if (t&1) r = r*(*this);return r;}};int main(){ll x,y,n;cin >> x >> y >> n;Matrix<mint> X(4,4);X[0][0] = x;X[0][1] = -5*y;X[1][0] = y;X[1][1] = x;X[2][0] = x;X[2][1] = -5*y;X[3][0] = y;X[3][1] = x;X[2][2] = 1;X[3][3] = 1;auto ret = X.pow(n-1);Matrix<mint> init(4,1);init[0][0] = x;init[1][0] = y;init[2][0] = x;init[3][0] = y;auto ans = ret*init;cout << ans[2][0].val() << " " << ans[3][0].val() << endl;return 0;}