#320433 (C++14) No.793 うし数列 2

• 問題: No.793 うし数列 2
• ユーザー: ldsyb
• 提出日時: 2019-03-02 01:15:54
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 950 bytes
• コンパイル時間: 1,826 ms
• コンパイル使用メモリ: 173,076 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-06-23 12:14:36
• 合計ジャッジ時間: 2,565 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

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

ソースコード

#include <bits/stdc++.h>
using namespace std;

using vec = vector<int64_t>;
using mat = vector<vec>;

int main()
{
	int64_t n, mod = 1000000007;
	cin >> n;

	auto mul = [](mat &a, mat &b, int64_t mod) -> mat {
		mat c(a.size(), vec(b[0].size(), 0));
		for (int i = 0; i < a.size(); i++)
		{
			for (int j = 0; j < b[0].size(); j++)
			{
				for (int k = 0; k < b.size(); k++)
				{
					(c[i][j] += a[i][k] * b[k][j] % mod) %= mod;
				}
			}
		}
		return c;
	};
	auto pow = [](mat &a, int64_t n, int64_t mod, auto &mul) -> mat {
		int m = a.size();
		mat r(m, vec(m, 0));
		for (int i = 0; i < m; i++)
		{
			r[i][i] = 1;
		}
		for (; n; n >>= 1)
		{
			if (n & 1)
			{
				r = mul(r, a, mod);
			}
			a = mul(a, a, mod);
		}
		return r;
	};

	mat a(3, vec(3)), r(3, vec(1));
	a[0][0] = 10;
	a[0][2] = 3;
	a[1][0] = 1;
	a[2][2] = 1;
	r[0][0] = 1;
	r[2][0] = 1;

	a = pow(a, n, mod, mul);
	r = mul(a, r, mod);

	cout << r[0][0] << endl;

	return 0;
}