結果

問題 No.2600 Avator Height
コンテスト
ユーザー amentorimaru
提出日時 2023-10-27 15:34:04
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 767 bytes
コンパイル時間 10,357 ms
コンパイル使用メモリ 141,456 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 23:57:15
合計ジャッジ時間 4,686 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other WA * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<iostream>
#include<string>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using mint = modint998244353;

vector<mint> mul(vector<mint>& a, vector<mint>& b) {
	vector<mint> res = { a[0] * b[0] + a[1] * b[2],a[0] * b[1] + a[1] * b[3],a[2] * b[0] + a[3] * b[2],a[2] * b[1] + a[3] * b[3] };
	return res;
}

vector<mint> power(long long n, vector<mint>& mut) {
	vector<mint> res = { 1,0,0,1 };
	if (n == 0)
		return res;
	auto pow = mul(mut, mut);
	res = power(n / 2, pow);
	if (n % 2)res = mul(res, mut);
	return res;
}

int main() {
	long long n;
	cin >> n;
	vector<mint> mut = { 0,1,1,1 };
	auto res = power(n - 1, mut);
	cout << (5 * (res[0] + res[1]) * (res[0] + res[1]) - (res[0] + 3 * res[1]) * (res[0] + 3 * res[1])).val() << endl;;
}
0