結果

問題 No.3226 2×2行列累乗
コンテスト
ユーザー Rumain831
提出日時 2025-08-16 18:28:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,140 bytes
コンパイル時間 961 ms
コンパイル使用メモリ 85,980 KB
実行使用メモリ 7,716 KB
最終ジャッジ日時 2025-08-16 18:28:07
合計ジャッジ時間 2,666 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
using ll = long long;

ll K;

struct Matrix{
  int n;
  vector<vector<ll>> a;
  Matrix(int n) :n(n), a(n, vector<ll>(n, 0)){}
  
  Matrix iden(int n){
    Matrix I(n);
    for(int i=0; i<n; i++) I.a[i][i]=1;
    return I;
  }

  Matrix operator*(const Matrix& right) const { //right*base
    Matrix ans(n);
    for(int i=0; i<n; i++){
      for(int k=0; k<n; k++){
        for(int j=0; j<n; j++){
          ans.a[i][j]+=a[i][k]*right.a[k][j];
          ans.a[i][j]=(ans.a[i][j]%K+K+K)%K;
        }
      }
    }
    return ans;
  }

};

Matrix pow(Matrix A, ll k){
  Matrix R(A.n);
  R=R.iden(A.n);
  while(k){
    if(k&1) R=A*R;
    A=A*A;
    k>>=1;
  }
  return R;
}

int main(void){
  Matrix A(2), powa(2);
  for(int i=0; i<2; i++)for(int j=0; j<2; j++) cin >> A.a[i][j];
  ll s, t, n; cin >> s >> t >> n >> K;
  powa=pow(A, n);
  //for(int i=0; i<2; i++)for(int j=0; j<2; j++) cout << powa.a[i][j] << (j==1?'\n':' ');
  cout << (powa.a[0][0]*s%K+K+powa.a[0][1]*t%K+K)%K << ' ' << (powa.a[1][0]*s%K+K+powa.a[1][1]*t%K+K)%K << endl;
  return 0;
}
0