結果

問題 No.891 隣接3項間の漸化式
ユーザー saxofone111saxofone111
提出日時 2021-03-25 13:08:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 6 ms / 2,000 ms
コード長 1,264 bytes
コンパイル時間 3,732 ms
コンパイル使用メモリ 201,188 KB
最終ジャッジ日時 2025-01-19 21:38:44
ジャッジサーバーID
(参考情報)
judge1 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};
using mat = vector<llvec>;

/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll a, b, n;
  cin >> a >> b >> n;
  mat e = {{1, 0}, {0, 1}};
  mat A = {{a, b}, {1, 0}};
  auto prod =[](mat a, mat b){
    mat ans(2, llvec(2, 0));
    rep(i, 2)rep(j, 2)rep(k, 2){
      ans[i][j] += a[i][k]*b[k][j];
      ans[i][j] %=MOD;
    }
    return ans;
  };
  if(n==1){
    cout << 1 << endl;
    return 0;
  }else if(n==0){
    cout << 0 << endl;
    return 0;
  }
  n-=1;
  while(n>0){
    if(n%2==1){
      e = prod(A, e);
    }
    A = prod(A, A);
    n/=2;
  }
  cout << (e[0][0])%MOD<<endl;
  return 0;
}
0