#991849 (C++23) No.2810 Have Another Go (Hard)

提出ソース

結果

問題	No.2810 Have Another Go (Hard)
ユーザー	highlighter
提出日時	2024-06-29 00:36:05
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	9,692 ms / 3,000 ms
コード長	3,425 bytes
コンパイル時間	5,631 ms
コンパイル使用メモリ	321,528 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-06-29 00:41:57
合計ジャッジ時間	347,149 ms
ジャッジサーバーID （参考情報）	judge4 / judge2

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 5,248 KB
testcase_01	AC	9,465 ms 5,248 KB
testcase_02	AC	9,480 ms 6,940 KB
testcase_03	AC	9,404 ms 5,376 KB
testcase_04	AC	9,525 ms 5,376 KB
testcase_05	AC	9,489 ms 5,376 KB
testcase_06	AC	9,478 ms 5,376 KB
testcase_07	AC	9,465 ms 5,376 KB
testcase_08	AC	9,527 ms 5,376 KB
testcase_09	AC	9,517 ms 5,376 KB
testcase_10	AC	9,483 ms 5,376 KB
testcase_11	AC	49 ms 5,376 KB
testcase_12	AC	52 ms 5,376 KB
testcase_13	AC	19 ms 5,376 KB
testcase_14	AC	66 ms 5,376 KB
testcase_15	AC	48 ms 5,376 KB
testcase_16	AC	2,041 ms 5,376 KB
testcase_17	AC	8,330 ms 5,376 KB
testcase_18	AC	5,199 ms 5,376 KB
testcase_19	AC	8,030 ms 5,376 KB
testcase_20	AC	1,223 ms 5,376 KB
testcase_21	AC	2,812 ms 5,376 KB
testcase_22	AC	5,175 ms 5,376 KB
testcase_23	AC	3,662 ms 5,376 KB
testcase_24	AC	5,959 ms 5,376 KB
testcase_25	AC	9,167 ms 5,376 KB
testcase_26	AC	9,587 ms 5,376 KB
testcase_27	AC	9,619 ms 5,376 KB
testcase_28	AC	9,614 ms 5,376 KB
testcase_29	AC	9,573 ms 5,376 KB
testcase_30	AC	9,609 ms 5,376 KB
testcase_31	AC	9,624 ms 5,376 KB
testcase_32	AC	9,470 ms 5,376 KB
testcase_33	AC	9,617 ms 5,376 KB
testcase_34	AC	9,656 ms 5,376 KB
testcase_35	AC	9,564 ms 5,376 KB
testcase_36	AC	9,599 ms 5,376 KB
testcase_37	AC	9,582 ms 5,376 KB
testcase_38	AC	9,588 ms 6,944 KB
testcase_39	AC	9,642 ms 5,376 KB
testcase_40	AC	9,609 ms 5,376 KB
testcase_41	AC	9,692 ms 5,376 KB
testcase_42	AC	9,555 ms 5,376 KB
testcase_43	AC	9,625 ms 5,376 KB
testcase_44	AC	9,654 ms 5,376 KB
testcase_45	AC	9,589 ms 5,376 KB
testcase_46	AC	6 ms 5,376 KB
testcase_47	AC	18 ms 5,376 KB
testcase_48	AC	38 ms 5,376 KB
testcase_49	AC	15 ms 5,376 KB
testcase_50	AC	24 ms 5,376 KB

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

ソースコード

raw source code

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using mint=modint998244353;

mint Bostan_Mori(long long N){
	vector<mint> Q={1,-1,-1,-1,-1,-1,-1};
	vector<mint> P={1,0,0,0,0,0};
	int d=6;
	while(N>0){
		vector<mint> Q_minus=Q;
		for(int i=1;i<(int)(Q.size());i+=2){
			Q_minus[i]*=-1;
		}
		vector<mint> V_sub=convolution(Q,Q_minus);
		Q.resize(((int)(V_sub.size())+1)/2);
		for(int i=0;i<(int)(Q.size());i++){
			Q[i]=V_sub[i*2];
		}
		vector<mint> U=convolution(P,Q_minus);
		if(N%2==0){
			P.resize(((int)(U.size())+1)/2);
			for(int i=0;i<(int)(U.size());i+=2){
				P[i/2]=U[i];
			}
			N/=2;
			continue;
		}
		P.resize((int)(U.size())/2);
		for(int i=1;i<(int)(U.size());i+=2){
			P[i/2]=U[i];
		}
		N/=2;
		continue;
	}
	return P[0]/Q[0];
}

template<typename T>
struct matrix{
	int H,W;
	T table[6][6];
	matrix(int h,int w) : H(h),W(w){
		for(int i=0;i<h;i++){
			for(int j=0;j<w;j++){
				table[i][j]=0;
			}
		}
	}
	matrix pow(long long N){
		matrix x=*this;
		matrix r(H,H);
		for(int i=0;i<H;i++){
			r.table[i][i]=1;
		}
		while(N){
			if(N&1)r*=x;
			x*=x;
			N/=2;
		}
		return r;
	}
	matrix operator*=(const matrix &other){
		int h=other.H;
		int w=other.W;
		//結果はH*w行列になる
		matrix result(H,w);
		for(int i=0;i<H;i++){
			for(int k=0;k<W;k++){
				for(int j=0;j<w;j++){
					result.table[i][j]+=table[i][k]*other.table[k][j];
				}
			}
		}
		*this=result;
		return *this;
	}
};

int main(){
	int k;
	long long N,M;
	scanf("%lld%lld%d",&N,&M,&k);
	mint W=0;
	for(int i=5;i>=0;i--){
		W+=Bostan_Mori(N*M+i)*(i-4);
	}
	for(;k--;){
		int C;
		scanf("%d",&C);
		matrix<mint> P(6,6);
		matrix<mint> Q(6,1);
		for(int i=0;i<6;i++){
			if(i==C){
				continue;
			}
			if(C<i){
				Q.table[i][0]=Bostan_Mori(i)-Bostan_Mori(C)*Bostan_Mori(i-C);
				continue;
			}
			Q.table[i][0]=Bostan_Mori(i);
		}
		for(int j=0;j<6;j++){
			mint P_sub=0;
			for(int i=5;i>=0;i--){
				if(i==C){
					continue;
				}
				if(j==C){
					continue;
				}
				if(i<C && j<C){
					P.table[j][i]=Bostan_Mori(N+j-i)-Bostan_Mori(C-i)*Bostan_Mori(N+j-C);
					P.table[j][i]-=P_sub;
					P_sub+=P.table[j][i]+P_sub;
					continue;
				}
				if(C<j && C<i){
					P.table[j][i]=Bostan_Mori(N+j-i)-Bostan_Mori(N+C-i)*Bostan_Mori(j-C);
					P.table[j][i]-=P_sub;
					P_sub+=P.table[j][i]+P_sub;
					continue;
				}
				if(i<C && C<j){
					mint res1=Bostan_Mori(C-i);
					mint res2=Bostan_Mori(N);
					mint res3=Bostan_Mori(j-C);
					mint res4=Bostan_Mori(N+C-i);
					mint res5=Bostan_Mori(N+j-C);
					mint res6=Bostan_Mori(N+j-i);
					P.table[j][i]=res6-res1*res5-res4*res3+res1*res2*res3;
					P.table[j][i]-=P_sub;
					P_sub+=P.table[j][i]+P_sub;
					continue;
				}
				P.table[j][i]=Bostan_Mori(N+j-i);
				P.table[j][i]-=P_sub;
				P_sub+=P.table[j][i]+P_sub;
			}
		}
		P=P.pow(M-1);
		P*=Q;
		mint R[6];
		for(int j=0;j<6;j++){
			R[j]=0;
			if(j==4){
				continue;
			}
			mint R_sub=0;
			for(int i=5;i>=0;i--){
				if(P.table[i][0]==0){
					continue;
				}
				if(i<C){
					mint res=Bostan_Mori(C-i)*Bostan_Mori(N+j-C);
					res=Bostan_Mori(N+j-i)-res;
					res-=R_sub;
					R_sub+=res+R_sub;
					R[j]+=res*P.table[i][0];
					continue;
				}
				mint res=Bostan_Mori(N+j-i);
				res-=R_sub;
				R_sub+=res+R_sub;
				R[j]+=res*P.table[i][0];
			}
		}
		mint ans=0;
		for(int i=5;i>=0;i--){
			ans+=R[i]*(i-4);
		}
		ans=W-ans;
		printf("%d\n",ans.val());
	}
}

yukicoder

結果

テストケース

ソースコード