結果

問題 No.2810 Have Another Go (Hard)
ユーザー highlighterhighlighter
提出日時 2024-06-28 21:00:24
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,295 bytes
コンパイル時間 847 ms
コンパイル使用メモリ 82,008 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-28 21:00:29
合計ジャッジ時間 4,989 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<stdio.h>
#include<unordered_map>
#include<atcoder/modint>
using namespace std;
using namespace atcoder;
using mint=modint998244353;

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

mint Bostan_Mori(long long N){
	if(mp.count(N)){
		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){
		copy(begin(Q_data[0]),end(Q_data[0]),begin(Q));
	}
	if(mem==0){
		Q[0]=1;
		for(int i=1;i<=6;i++){
			Q[i]=-1;
		}
		copy(begin(Q),end(Q),begin(Q_data[0]));
		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];
				}
			}
			copy(begin(nQ),end(nQ),begin(Q_data[mem]));
			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];
				}
			}
			copy(begin(nP),end(nP),begin(P));
		}
		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];
				}
			}
			copy(begin(nP),end(nP),begin(P));
		}
		copy(begin(Q_data[count]),end(Q_data[count]),begin(Q));
		N/=2;
		continue;
	}
	mp[M]=P[0];
	return P[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(){
	mp.reserve(1020);
	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());
	}
	printf("%d\n",(int)(mp.size()));
}
0