#493428 (C++14) No.1073 無限すごろく

• 問題: No.1073 無限すごろく
• ユーザー: platinum
• 提出日時: 2020-06-06 15:17:20
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 899 bytes
• コンパイル時間: 1,633 ms
• コンパイル使用メモリ: 174,888 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-06 02:05:15
• 合計ジャッジ時間: 2,866 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 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	1 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	1 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	1 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	1 ms; 5,376 KB
testcase_17	AC	1 ms; 5,376 KB
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	2 ms; 5,376 KB
testcase_20	AC	2 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	2 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	1 ms; 5,376 KB
testcase_29	AC	2 ms; 5,376 KB
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	1 ms; 5,376 KB
testcase_32	AC	2 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
#define rep(i,n) for(int i=0; i<(int)(n); i++)

using namespace std;
using LL = long long;
using mat = vector<vector<LL>>;
const LL mod=1e9+7;

mat matmul(mat &A, mat &B){
	mat C(A.size(),vector<LL>(B[0].size()));
	rep(i,A.size()){
		rep(k,B.size()){
			rep(j, B[0].size()){
				(C[i][j]+=A[i][k]*B[k][j])%=mod;
			}
		}
	}
	return C;
}

mat matpow(mat A, LL n){
	mat C(A.size(),vector<LL>(A.size()));
	rep(i,A.size()) C[i][i]=1;
	while(n>0){
		if(n&1) C=matmul(C,A);
		A=matmul(A,A);
		n>>=1;
	}
	return C;
}

LL modinv(LL x){
	LL m=mod, u=1, v=0;
	while(m){
		LL t=x/m;
		x-=t*m; swap(x,m);
		u-=t*v; swap(u,v);
	}
	u%=mod;
	if(u<0) u+=mod;
	return u;
}

int main(){
	LL N;
	cin >> N;
	mat A(6,vector<LL>(6));
	rep(i,6) A[5][i]=modinv(6);
	rep(i,5){
		rep(j,6){
			if(i+1==j) A[i][j]=1;
			else A[i][j]=0;
		}
	}
	mat B=matpow(A,N);
	cout << B[5][5] << endl;

	return 0;
}