#565934 (C++14) No.720 行列のできるフィボナッチ数列道場 (2)

• 問題: No.720 行列のできるフィボナッチ数列道場 (2)
• ユーザー: 👑 Nachia
• 提出日時: 2020-10-11 18:05:13
• 言語: C++14 (gcc 12.3.0 + boost 1.83.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

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	2 ms; 5,376 KB
testcase_18	AC	2 ms; 5,376 KB
testcase_19	AC	2 ms; 5,376 KB
testcase_20	AC	2 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB
testcase_22	AC	2 ms; 5,376 KB

ソースコード

#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;
}