結果

問題 No.194 フィボナッチ数列の理解(1)
ユーザー roiti46roiti46
提出日時 2015-10-11 04:45:03
言語 C++11
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,616 bytes
コンパイル時間 1,713 ms
コンパイル使用メモリ 166,904 KB
実行使用メモリ 11,520 KB
最終ジャッジ日時 2024-07-21 06:17:56
合計ジャッジ時間 2,336 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<ll> vec;
typedef vector<vec> mat;

#define REP(i, a, b) for (int i = (int)(a); i < (int)(b); i++)
#define rep(i, a) REP(i, 0, a)
#define EACH(i, a) for (auto i: a)
#define ITR(x, a) for (__typeof(a.begin()) x = a.begin(); x != a.end(); x++)
#define ALL(a) (a.begin()), (a.end())
#define HAS(a, x) (a.find(x) != a.end())

ll N, K;
const ll mod = 1000000007;
ll F[10006], S[1000005];

mat mat_mul(mat &A, mat &B) {
  mat C(A.size(), vec(B[0].size()));
  for (int i = 0; i < A.size(); i++)
    for (int k = 0; k < B.size(); k++)
      for (int j = 0; j < B[0].size(); j++)
        C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % mod;
  return C;
}

mat mat_pow(mat A, ll n) {
  mat B(A.size(), vec(A.size()));
  for (int i = 0; i < A.size(); i++) B[i][i] = 1;
  while (n) {
    if (n & 1) B = mat_mul(B, A);
    A = mat_mul(A, A);
    n >>= 1;
  }
  return B;
}

void solve1() {
  S[0] = F[0];
  REP(i, 1, N) S[i] = (S[i - 1] + F[i]) % mod;
  REP(i, N, K + 1) {
    S[i] = (2 * S[i - 1] - S[i - N - 1] + mod) % mod;
  }
  cout << (S[K - 1] - S[K - N - 1] + mod) % mod << " " << S[K - 1] << endl;
}


void solve2() {
  mat B(N + 1, vec(1));
  rep(i, N) B[N - i][0] = F[i];
  rep(i, N) B[0][0] += F[i];
  mat A(N + 1, vec(N + 1));
  rep(i, N + 1) A[0][i] = 1;
  rep(i, N) A[1][i + 1] = 1;
  rep(i, N - 1) A[i + 2][i + 1] = 1;
  A = mat_pow(A, K - N);
  B = mat_mul(A, B);
  cout << B[1][0] << " " << B[0][0] << endl;
}

int main(void) {
  cin >> N >> K;
  rep(i, N) cin >> F[i];

  if (N > 30)
    solve1();
  else
    solve2();

  return 0;
}
0