結果

問題 No.720 行列のできるフィボナッチ数列道場 (2)
ユーザー 👑 Nachia
提出日時 2020-10-11 18:05:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 859 bytes
コンパイル時間 2,538 ms
コンパイル使用メモリ 166,916 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 17:32:14
合計ジャッジ時間 2,965 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using LL = long long;
using ULL = unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

const ULL M = 1000000007;

const int matrix_size = 3;

struct Matrix {
	ULL X[matrix_size][matrix_size] = {};
	static Matrix id() {
		Matrix res;
		rep(i, matrix_size) res.X[i][i] = 1;
		return res;
	}
};

Matrix operator*(Matrix l, Matrix r) {
	Matrix res;
	rep(i, matrix_size) rep(j, matrix_size) rep(k, matrix_size)
		res.X[i][j] = (res.X[i][j] + l.X[i][k] * r.X[k][j]) % M;
	return res;
}

Matrix operator^(Matrix l, ULL i) {
	if (i == 0) return Matrix::id();
	auto res = (l * l) ^ (i / 2);
	if (i % 2 == 1) res = res * l;
	return res;
}

int main() {
	ULL N, I; cin >> N >> I;

	Matrix X = { { {1,1,0},{1,0,0},{0,0,1} } };
	X = X ^ I;

	X.X[0][2] = 1;
	X = X ^ (N + 1);

	cout << X.X[1][2] << endl;

	return 0;
}
0