結果
問題 | No.3006 ベイカーの問題 |
ユーザー |
|
提出日時 | 2025-01-17 21:59:07 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 3 ms / 2,000 ms |
コード長 | 1,637 bytes |
コンパイル時間 | 7,633 ms |
コンパイル使用メモリ | 331,980 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2025-01-17 21:59:35 |
合計ジャッジ時間 | 7,277 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/all>using mint = atcoder::static_modint<998244353>;//using mint = atcoder::static_modint<1000000007>;using namespace std;using namespace atcoder;using ld = long double;using ll = long long;#define mp(a,b) make_pair(a,b)#define rep(i,s,n) for(int i=s; i<n; i++)const vector<int> dx{1,0,-1,0},dy{0,1,0,-1};vector<vector<mint>> A;vector<vector<mint>> arrayproduct(vector<vector<mint>>&X,vector<vector<mint>>&Y){int arraysize=X.size();vector<vector<mint>> output(arraysize,vector<mint>(arraysize,0));rep(i,0,arraysize)rep(j,0,arraysize){rep(k,0,arraysize)output[i][j]+=X[i][k]*Y[k][j];}return output;}vector<vector<mint>> arraypower(vector<vector<mint>>&X,ll z){if(z==1)return X;ll half=z/2;auto H=arraypower(X,half);auto U=arrayproduct(H,H);if(z%2==0)return U;else return arrayproduct(U,X);}ll x,y;vector<mint> solve(ll n){if(n==1)return {x,y};ll h=n/2;auto H=solve(h);if(n&1){auto P=arraypower(A,h+1);vector<mint> out(2);rep(i,0,2){rep(j,0,2){out[j]+=P[i][j]*H[i];out[j]+=A[i][j]*H[i];}}out[0]+=x;out[1]+=y;return out;}else{auto P=arraypower(A,h);vector<mint> out=H;rep(i,0,2){rep(j,0,2){out[j]+=P[i][j]*H[i];}}return out;}}int main(){cin >> x >> y;ll n;cin >> n;A={{x,y},{-5*y,x}};auto ans=solve(n);rep(i,0,2)cout << ans[i].val() << " ";}