結果

問題 No.891 隣接3項間の漸化式
ユーザー le_panda_noirle_panda_noir
提出日時 2020-02-21 22:37:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 795 bytes
コンパイル時間 751 ms
コンパイル使用メモリ 67,324 KB
最終ジャッジ日時 2025-01-09 01:39:39
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
using namespace std;
using ll = long long;
const int MOD = 1e9+7;
#define rep(i,n) for(int i=0;i<(n);++i)

struct Matrix {
  ll a, b, c, d;
public:
  Matrix(ll a, ll b, ll c, ll d) : a(a % MOD), b(b % MOD), c(c % MOD), d(d % MOD) {}
};

Matrix mul(Matrix A, Matrix B) {
  return Matrix(A.a * B.a + A.b * B.c, A.a * B.b + A.b * B.d, A.c * B.a + A.d * B.c, A.c * B.b + A.d * B.d);
}
Matrix pow(Matrix A, int p) {
  if (p == 1)
    return A;
  if (p % 2 == 0)
    return pow(mul(A, A), p / 2);
  else
    return mul(pow(mul(A, A), p / 2), A);
}

int main() {
  int a, b, n;
  cin >> a >>  b >>  n;

  if (n == 0)
    cout << 0 << endl;
  else if (n == 1)
    cout << 1 << endl;
  else {
    Matrix m(0, 1, b, a);
    m = pow(m, n-1);
    cout << m.d << endl;
  }

  return 0;
}
0