結果

問題 No.194 フィボナッチ数列の理解(1)
コンテスト
ユーザー sugim48
提出日時 2015-04-26 23:42:47
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 203 ms / 5,000 ms
コード長 2,153 bytes
コンパイル時間 840 ms
コンパイル使用メモリ 99,560 KB
実行使用メモリ 11,088 KB
最終ジャッジ日時 2024-07-05 02:53:43
合計ジャッジ時間 4,368 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <algorithm>
#include <cstdio>
#include <functional>
#include <iostream>
#include <cfloat>
#include <climits>
#include <cstring>
#include <cmath>
#include <fstream>
#include <map>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <time.h>
#include <vector>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> i_i;
typedef pair<ll, int> ll_i;
typedef pair<double, int> d_i;
typedef pair<ll, ll> ll_ll;
typedef pair<double, double> d_d;
struct edge { int u, v; ll w; };

int INF = INT_MAX / 10;
ll MOD = 1000000007;
ll _MOD = 1000000009;
double EPS = 1e-10;

vector< vector<ll> > matmul(vector< vector<ll> >& A, vector< vector<ll> >& B) {
	int n = A.size();
	vector< vector<ll> > C(n, vector<ll>(n));
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			for (int k = 0; k < n; k++)
				C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;
	return C;
}

vector< vector<ll> > matpow(vector< vector<ll> > A, ll x) {
	int n = A.size();
	vector< vector<ll> > B(n, vector<ll>(n));
	for (int i = 0; i < n; i++) B[i][i] = 1;
	while (x > 0) {
		if (x % 2) B = matmul(B, A);
		x /= 2;
		A = matmul(A, A);
	}
	return B;
}

int main() {
	int N; ll K; cin >> N >> K;
	vector<ll> A(N);
	for (int i = 0; i < N; i++)
		cin >> A[i];
	if (N <= 30) {
		vector<vector<ll> > a(N * 2, vector<ll>(N * 2));
		for (int i = 0; i < N; i++)
			a[0][i] = a[N + i][i] = a[N + i][N + i] = 1;
		for (int i = 0; i + 1 < N; i++)
			a[i + 1][i] = 1;
		vector<vector<ll> > b = matpow(a, K - 1);
		ll f = 0, s = 0;
		for (int i = 0; i < N; i++) {
			f = (f + b[N - 1][i] * A[N - i - 1]) % MOD;
			s = (s + b[N * 2 - 1][i] * A[N - i - 1]) % MOD;
		}
		cout << f << ' ' << (f + s) % MOD << endl;
	}
	else {
		vector<ll> a(K);
		ll sum = 0, s = 0;
		for (int i = 0; i < N; i++) {
			a[i] = A[i];
			sum = (sum + a[i]) % MOD;
		}
		for (int i = N; i < K; i++) {
			a[i] = sum;
			sum = (sum + a[i] - a[i - N]) % MOD;	
		}
		for (int i = 0; i < K; i++)
			s = (s + a[i]) % MOD;
		cout << (a[K - 1] + MOD) % MOD << ' ' << (s + MOD) % MOD << endl;
	}
}
0