結果

問題 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() << " ";
}
0