結果

問題 No.2810 Have Another Go (Hard)
ユーザー highlighterhighlighter
提出日時 2024-07-07 15:22:45
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 471 ms / 3,000 ms
コード長 4,552 bytes
コンパイル時間 2,924 ms
コンパイル使用メモリ 178,548 KB
実行使用メモリ 7,716 KB
最終ジャッジ日時 2024-07-07 15:23:32
合計ジャッジ時間 18,102 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 281 ms
7,636 KB
testcase_02 AC 281 ms
7,536 KB
testcase_03 AC 281 ms
7,632 KB
testcase_04 AC 302 ms
7,632 KB
testcase_05 AC 276 ms
7,632 KB
testcase_06 AC 280 ms
7,520 KB
testcase_07 AC 276 ms
7,612 KB
testcase_08 AC 298 ms
7,540 KB
testcase_09 AC 277 ms
7,536 KB
testcase_10 AC 285 ms
7,540 KB
testcase_11 AC 3 ms
5,376 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 4 ms
5,376 KB
testcase_15 AC 3 ms
5,376 KB
testcase_16 AC 80 ms
5,376 KB
testcase_17 AC 331 ms
7,636 KB
testcase_18 AC 230 ms
7,636 KB
testcase_19 AC 315 ms
7,636 KB
testcase_20 AC 49 ms
5,376 KB
testcase_21 AC 111 ms
5,376 KB
testcase_22 AC 199 ms
7,540 KB
testcase_23 AC 143 ms
5,376 KB
testcase_24 AC 267 ms
7,536 KB
testcase_25 AC 351 ms
7,600 KB
testcase_26 AC 393 ms
7,632 KB
testcase_27 AC 389 ms
7,612 KB
testcase_28 AC 400 ms
7,716 KB
testcase_29 AC 395 ms
7,636 KB
testcase_30 AC 387 ms
7,636 KB
testcase_31 AC 425 ms
7,632 KB
testcase_32 AC 400 ms
7,640 KB
testcase_33 AC 391 ms
7,632 KB
testcase_34 AC 404 ms
7,696 KB
testcase_35 AC 391 ms
7,640 KB
testcase_36 AC 387 ms
7,548 KB
testcase_37 AC 407 ms
7,636 KB
testcase_38 AC 387 ms
7,636 KB
testcase_39 AC 384 ms
7,636 KB
testcase_40 AC 471 ms
7,636 KB
testcase_41 AC 388 ms
7,608 KB
testcase_42 AC 417 ms
7,636 KB
testcase_43 AC 386 ms
7,612 KB
testcase_44 AC 383 ms
7,620 KB
testcase_45 AC 408 ms
7,632 KB
testcase_46 AC 2 ms
5,376 KB
testcase_47 AC 2 ms
5,376 KB
testcase_48 AC 2 ms
5,376 KB
testcase_49 AC 1 ms
5,376 KB
testcase_50 AC 2 ms
5,376 KB
testcase_51 AC 39 ms
5,376 KB
testcase_52 AC 26 ms
5,376 KB
testcase_53 AC 59 ms
5,376 KB
testcase_54 AC 78 ms
5,376 KB
testcase_55 AC 27 ms
5,376 KB
testcase_56 AC 70 ms
5,376 KB
testcase_57 AC 49 ms
5,376 KB
testcase_58 AC 38 ms
5,376 KB
testcase_59 AC 95 ms
5,376 KB
testcase_60 AC 59 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<atcoder/modint>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace atcoder;
using namespace __gnu_pbds;
using mint=modint998244353;

int mem=0;
mint Q_data[62][7];
gp_hash_table<long long,mint> mp;

mint Bostan_Mori(long long N){
	if(mp.find(N)!=mp.end()){
		return mp[N];
	}
	if(N==0){
		mint ans=1;
		mp[N]=ans;
		return ans;
	}
	if(N<=6){
		mint ans=mint(2).pow(N-1);
		mp[N]=ans;
		return ans;
	}
	long long M=N;
	mint Q[7];
	if(mem>0){
		for(int i=0;i<7;i++){
			Q[i]=Q_data[0][i];
		}
	}
	if(mem==0){
		Q[0]=1;
		for(int i=1;i<=6;i++){
			Q[i]=-1;
		}
		for(int i=0;i<7;i++){
			Q_data[0][i]=Q[i];
		}
		mem++;
	}
	mint P[6];
	P[0]=1;
	for(int i=1;i<6;i++){
		P[i]=0;
	}
	int count=0;
	while(N>0){
		count++;
		if(mem<=count){
			mint nQ[7];
			for(int i=0;i<=6;i++){
				nQ[i]=0;
				for(int j=max(0,2*i-6);j<=min(2*i,6);j++){
					if(j%2==0){
						nQ[i]+=Q[j]*Q[2*i-j];
						continue;
					}
					nQ[i]-=Q[j]*Q[2*i-j];
				}
			}
			for(int i=0;i<7;i++){
				Q_data[mem][i]=nQ[i];
			}
			mem++;
		}
		mint nP[6];
		if(N%2==0){
			for(int i=0;i<12;i+=2){
				nP[i/2]=0;
				for(int j=max(0,i-5);j<=min(6,i);j++){
					if(j%2==0){
						nP[i/2]+=Q[j]*P[i-j];
						continue;
					}
					nP[i/2]-=Q[j]*P[i-j];
				}
			}
			for(int i=0;i<6;i++){
				P[i]=nP[i];
			}
		}
		if(N%2==1){
			for(int i=1;i<12;i+=2){
				nP[i/2]=0;
				for(int j=max(0,i-5);j<=min(6,i);j++){
					if(j%2==0){
						nP[i/2]+=Q[j]*P[i-j];
						continue;
					}
					nP[i/2]-=Q[j]*P[i-j];
				}
			}
			for(int i=0;i<6;i++){
				P[i]=nP[i];
			}
		}
		for(int i=0;i<7;i++){
			Q[i]=Q_data[count][i];
		}
		N/=2;
		continue;
	}
	mp[M]=P[0];
	return P[0];
}

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--){
		if(i==4){
			continue;
		}
		W+=Bostan_Mori(N*M+i)*(i-4);
	}
	M--;
	long long M_sub=M;
	mint mul[6][6];
	mint mul_sub[6][6];
	mint P[6][6];
	mint Q[6];
	mint R[6];
	for(;k--;){
		M=M_sub;
		int C;
		scanf("%d",&C);
		for(int i=0;i<6;i++){
			if(i==C){
				Q[i]=0;
				continue;
			}
			if(C<i){
				Q[i]=Bostan_Mori(i)-Bostan_Mori(C)*Bostan_Mori(i-C);
				continue;
			}
			Q[i]=Bostan_Mori(i);
		}
		for(int j=0;j<6;j++){
			mint P_sub=0;
			for(int i=5;i>=0;i--){
				if(i==C){
					P[j][i]=0;
					continue;
				}
				if(j==C){
					P[j][i]=0;
					continue;
				}
				if(i<C && j<C){
					P[j][i]=Bostan_Mori(N+j-i)-Bostan_Mori(C-i)*Bostan_Mori(N+j-C);
					P[j][i]-=P_sub;
					P_sub+=P[j][i]+P_sub;
					continue;
				}
				if(C<j && C<i){
					P[j][i]=Bostan_Mori(N+j-i)-Bostan_Mori(N+C-i)*Bostan_Mori(j-C);
					P[j][i]-=P_sub;
					P_sub+=P[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[j][i]=res6-res1*res5-res4*res3+res1*res2*res3;
					P[j][i]-=P_sub;
					P_sub+=P[j][i]+P_sub;
					continue;
				}
				P[j][i]=Bostan_Mori(N+j-i);
				P[j][i]-=P_sub;
				P_sub+=P[j][i]+P_sub;
			}
		}
		for(int i=0;i<6;i++){
			for(int j=0;j<6;j++){
				if(i==j){
					mul[i][j]=1;
				}
				else{
					mul[i][j]=0;
				}
			}
		}
		while(M){
			if(M&1){
				for(int i=0;i<6;i++){
					for(int j=0;j<6;j++){
						mul_sub[i][j]=0;
					}
				}
				for(int i=0;i<6;i++){
					for(int k=0;k<6;k++){
						for(int j=0;j<6;j++){
							mul_sub[i][j]+=mul[i][k]*P[k][j];
						}
					}
				}
				for(int i=0;i<6;i++){
					for(int j=0;j<6;j++){
						mul[i][j]=mul_sub[i][j];
					}
				}
			}
			for(int i=0;i<6;i++){
				for(int j=0;j<6;j++){
					mul_sub[i][j]=0;
				}
			}
			for(int i=0;i<6;i++){
				for(int k=0;k<6;k++){
					for(int j=0;j<6;j++){
						mul_sub[i][j]+=P[i][k]*P[k][j];
					}
				}
			}
			for(int i=0;i<6;i++){
				for(int j=0;j<6;j++){
					P[i][j]=mul_sub[i][j];
				}
			}
			M>>=1;
		}
		for(int i=0;i<6;i++){
			P[i][0]=0;
			for(int j=0;j<6;j++){
				P[i][0]+=mul[i][j]*Q[j];
			}
		}
		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[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[i][0];
					continue;
				}
				mint res=Bostan_Mori(N+j-i);
				res-=R_sub;
				R_sub+=res+R_sub;
				R[j]+=res*P[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());
	}
}
0