#567558 (C++17) No.1258 コインゲーム

提出ソース

結果

問題	No.1258 コインゲーム
ユーザー	沙耶花
提出日時	2020-10-16 21:49:33
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	1,181 ms / 2,000 ms
コード長	2,425 bytes
コンパイル時間	2,965 ms
コンパイル使用メモリ	217,968 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-07-20 21:26:26
合計ジャッジ時間	37,724 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	219 ms 5,248 KB
testcase_01	AC	276 ms 5,376 KB
testcase_02	AC	99 ms 5,376 KB
testcase_03	AC	932 ms 5,376 KB
testcase_04	AC	1,041 ms 5,376 KB
testcase_05	AC	164 ms 5,376 KB
testcase_06	AC	740 ms 5,376 KB
testcase_07	AC	77 ms 5,376 KB
testcase_08	AC	63 ms 5,376 KB
testcase_09	AC	371 ms 5,376 KB
testcase_10	AC	899 ms 5,376 KB
testcase_11	AC	1,015 ms 5,376 KB
testcase_12	AC	742 ms 5,376 KB
testcase_13	AC	221 ms 5,376 KB
testcase_14	AC	71 ms 5,376 KB
testcase_15	AC	875 ms 5,376 KB
testcase_16	AC	174 ms 5,376 KB
testcase_17	AC	1,000 ms 5,376 KB
testcase_18	AC	368 ms 5,376 KB
testcase_19	AC	331 ms 5,376 KB
testcase_20	AC	737 ms 5,376 KB
testcase_21	AC	975 ms 5,376 KB
testcase_22	AC	660 ms 5,376 KB
testcase_23	AC	488 ms 5,376 KB
testcase_24	AC	178 ms 5,376 KB
testcase_25	AC	466 ms 5,376 KB
testcase_26	AC	410 ms 5,376 KB
testcase_27	AC	916 ms 5,376 KB
testcase_28	AC	241 ms 5,376 KB
testcase_29	AC	262 ms 5,376 KB
testcase_30	AC	1,163 ms 5,376 KB
testcase_31	AC	274 ms 5,376 KB
testcase_32	AC	120 ms 5,376 KB
testcase_33	AC	758 ms 5,376 KB
testcase_34	AC	947 ms 5,376 KB
testcase_35	AC	318 ms 5,376 KB
testcase_36	AC	911 ms 5,376 KB
testcase_37	AC	356 ms 5,376 KB
testcase_38	AC	887 ms 5,376 KB
testcase_39	AC	239 ms 5,376 KB
testcase_40	AC	1,153 ms 5,376 KB
testcase_41	AC	1,134 ms 5,376 KB
testcase_42	AC	1,151 ms 5,376 KB
testcase_43	AC	1,161 ms 5,376 KB
testcase_44	AC	1,150 ms 5,376 KB
testcase_45	AC	1,138 ms 5,376 KB
testcase_46	AC	1,156 ms 5,376 KB
testcase_47	AC	1,181 ms 5,376 KB
testcase_48	AC	1,170 ms 5,376 KB
testcase_49	AC	1,159 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace atcoder;
using mint = modint1000000007;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000001

template <typename T,typename F0,typename F1>
struct matrix{
	F0 func0;
	F1 func1;
	vector<vector<T>> value;
	int height,width;
	T init_value0,init_value1;
	
	matrix(vector<vector<T>> X,F0 f0,F1 f1,T iv0,T iv1):func0(f0),func1(f1){
		height = X.size();
		width = X[0].size();
		value = X;
		init_value0 = iv0,init_value1 = iv1;
	}
	
	matrix(int h,int w,F0 f0,F1 f1,T iv0,T iv1):func0(f0),func1(f1){
		vector<vector<T>> d(h,vector<T>(w,iv0));
		height = h,width = w;
		value = d;
		init_value0 = iv0,init_value1 = iv1;
	}
	
	
	void set_1(){
		for(int i=0;i<height;i++){
			for(int j=0;j<width;j++){
				if(i==j)value[i][j] = init_value1;
				else value[i][j]=init_value0;
			}
		}
	}
	
	void show(){
		for(int i=0;i<height;i++){
			for(int j=0;j<width;j++){
				if(j!=0)cout<<' ';
				cout<<value[i][j];
			}
			cout<<endl;
		}
	}
	
	matrix &operator=(const matrix &another){
		value = another.value;
		height = another.height;
		width = another.width;
		return *this;
	}
	
	matrix &operator*=(const matrix &another){
		matrix<T,decltype(func0),decltype(func1)> R(height,another.width,func0,func1,init_value0,init_value1);
		for(int i=0;i<R.value.size();i++){
			for(int j=0;j<R.value[i].size();j++){
				R.value[i][j]=init_value0;
				for(int k=0;k<width;k++){
					R.value[i][j] = func0(R.value[i][j],func1(value[i][k],another.value[k][j]));
				}
			}
		}

		value = R.value;
		return (*this);
	}
	
	matrix operator*(const matrix &another)const{
		return (matrix(*this)*=another);
	}
	
	matrix beki(long long cnt){
		matrix<T,decltype(func0),decltype(func1)> R(height,width,func0,func1,init_value0,init_value1);
		R.set_1();
		auto temp = *this;
		while(cnt!=0LL){
			if(cnt%2==1){
				R *= temp;
			}
			temp *= temp;
			cnt/=2;
		}
		return R;
	}
};

int main(){	
	
	int t;
	cin>>t;
	
	auto f0 = [](mint a,mint b){
		return a+b;
	};
	
	auto f1 = [](mint a,mint b){
		return a*b;
	};
	
	rep(_,t){
		int N,M,X;
		cin>>N>>M>>X;
		vector<vector<mint>> temp = {{1,M},{M,1}};
		matrix<mint,decltype(f0),decltype(f1)> A(temp,f0,f1,0,1);
		
		A = A.beki(N);

		temp = {{1,0}};
		matrix<mint,decltype(f0),decltype(f1)> B(temp,f0,f1,0,1);
		B *= A;
		
		cout<<B.value[0][X].val()<<endl;
	}
			
	
    return 0;
}

yukicoder

結果

テストケース

ソースコード