結果

問題 No.718 行列のできるフィボナッチ数列道場 (1)
コンテスト
ユーザー pekempey
提出日時 2018-07-27 23:56:43
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,132 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 566 ms
コンパイル使用メモリ 71,376 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-05 16:23:51
合計ジャッジ時間 1,267 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <algorithm>
#include <vector>
#include <array>

using namespace std;

const int mod = 1e9 + 7;

struct Modint {
  int n;
  Modint(int n = 0) : n(n) {}
};

Modint operator+(Modint a, Modint b) { return (a.n += b.n) >= mod ? a.n - mod : a.n; }
Modint operator-(Modint a, Modint b) { return (a.n -= b.n) < 0 ? a.n + mod : a.n; }
Modint operator*(Modint a, Modint b) { return 1LL * a.n * b.n % mod; }
Modint &operator+=(Modint &a, Modint b) { return a = a + b; }
Modint &operator-=(Modint &a, Modint b) { return a = a - b; }
Modint &operator*=(Modint &a, Modint b) { return a = a * b; }

// ax + b
using P = pair<Modint, Modint>;

P mul(P p, P q) {
  // p * q = ax^2 + bx + c = a(x + 1) + bx + c
  Modint a = p.first * q.first;
  Modint b = p.first * q.second + p.second * q.first;
  Modint c = p.second * q.second;
  return {a + b, a + c};
}

Modint fib(long long x) {
  P res(0, 1);
  P a(1, 0);
  while (x > 0) {
    if (x & 1) {
      res = mul(res, a);
    }
    a = mul(a, a);
    x >>= 1;
  }
  return res.first;
}

int main() {
  long long n;
  cin >> n;
  cout << (fib(n) * fib(n + 1)).n << endl;
}
0