結果

問題 No.3006 ベイカーの問題
ユーザー makichan
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#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() << " ";
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0