結果

問題 No.534 フィボナッチフィボナッチ数
ユーザー treeonetreeone
提出日時 2017-06-24 00:05:18
言語 C++11
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,816 bytes
コンパイル時間 1,540 ms
コンパイル使用メモリ 165,136 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-03 03:55:31
合計ジャッジ時間 2,268 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, a, n) for(int i = a; i < n; i++)
#define repb(i, a, b) for(int i = a; i >= b; i--)
#define all(a) a.begin(), a.end()
#define o(a) cout << a << endl
#define int long long
#define fi first
#define se second
using namespace std;
typedef pair<int, int> P;

// typedef double number;
typedef int number;
const number eps = 1e-8;
typedef vector<number> ARRAY;
const int mod0 = 1e9 + 7;
typedef vector<ARRAY> matrix;

// O( n )
matrix identity(int n) {
  matrix A(n, ARRAY(n));
  for (int i = 0; i < n; ++i) A[i][i] = 1;
  return A;
}
// O( n )
number inner_product(const ARRAY &a, const ARRAY &b) {
  number ans = 0;
  for (int i = 0; i < a.size(); ++i)
    ans += a[i] * b[i];
  return ans;
}
// O( n^2 )
ARRAY mul(const matrix &A, const ARRAY &x) {
  ARRAY y(A.size());
  for (int i = 0; i < A.size(); ++i)
    for (int j = 0; j < A[0].size(); ++j)
      y[i] = A[i][j] * x[j];
  return y;
}
// O( n^3 )
matrix mul(const matrix &A, const matrix &B, const int mod) {
  matrix C(A.size(), ARRAY(B[0].size()));
  for (int i = 0; i < C.size(); ++i)
    for (int j = 0; j < C[i].size(); ++j)
      for (int k = 0; k < A[i].size(); ++k){
        C[i][j] += (A[i][k] * B[k][j] % mod);
        C[i][j] %= mod;
      }
  return C;
}
// O( n^3 log e )
matrix pow(const matrix &A, int e, const int mod) {
  return e == 0 ? identity(A.size())  :
     e % 2 == 0 ? pow(mul(A, A, mod), e/2, mod) : mul(A, pow(A, e-1, mod), mod);
}

signed main(){
    int n;
    cin >> n;
    vector<vector<int> > A(2, vector<int>(2, 0));
    A[0][0] = A[0][1] = A[1][0] = 1;
    A[1][1] = 0;
    vector<vector<int> > B = A;
    int mod1 = 2 * mod0 + 2;
    B = pow(A, n, mod1);
    B[1][0] = (B[1][0] + mod1) % mod1;
    cout << B[1][0] << endl;
    // A = pow(A, B[1][0], mod0);
    cout << A[1][0] << endl;
}
0