結果

問題 No.2530 Yellow Cards
ユーザー shobonvipshobonvip
提出日時 2023-10-25 01:50:23
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 858 bytes
コンパイル時間 2,420 ms
コンパイル使用メモリ 213,892 KB
実行使用メモリ 4,348 KB
最終ジャッジ日時 2023-10-25 01:50:27
合計ジャッジ時間 3,530 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 2 ms
4,348 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 2 ms
4,348 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 2 ms
4,348 KB
testcase_09 AC 2 ms
4,348 KB
testcase_10 AC 2 ms
4,348 KB
testcase_11 AC 1 ms
4,348 KB
testcase_12 AC 2 ms
4,348 KB
testcase_13 AC 2 ms
4,348 KB
testcase_14 AC 2 ms
4,348 KB
testcase_15 AC 2 ms
4,348 KB
testcase_16 AC 2 ms
4,348 KB
testcase_17 AC 2 ms
4,348 KB
testcase_18 AC 2 ms
4,348 KB
testcase_19 AC 2 ms
4,348 KB
testcase_20 AC 2 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include<atcoder/modint>
typedef atcoder::modint998244353 mint;

int main(){
	long long n; cin >> n;
	long long k; cin >> k;
	mint ninv = mint(n).inv();
	vector mat(60, vector(3, vector<mint>(3)));
	mat[0][0][0] = mint(3*n-2)*ninv;
	mat[0][0][1] = -mint(3*n-4)*ninv;
	mat[0][0][2] = mint(n-2)*ninv;
	mat[0][1][0] = 1;
	mat[0][2][1] = 1;
	for (int num=0; num<59; num++){
		for (int i=0; i<3; i++){
			for (int j=0; j<3; j++){
				for (int k=0; k<3; k++){
					mat[num+1][i][j] += mat[num][i][k] * mat[num][k][j];
				}
			}
		}
	}
	vector<mint> a = {ninv, 0, 0};
	for (int num=0; num<60; num++){
		if (k >> num & 1){
			vector<mint> na = {0, 0, 0};
			for (int i=0; i<3; i++){
				for (int j=0; j<3; j++){
					na[i] += mat[num][i][j] * a[j];
				}
			}
			a = na;
		}
	}
	cout << (a[2] + mint(n)).val() << endl;
}
0