結果

問題 No.214 素数サイコロと合成数サイコロ (3-Medium)
ユーザー 沙耶花沙耶花
提出日時 2024-11-06 00:33:43
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 389 ms / 3,000 ms
コード長 1,814 bytes
コンパイル時間 5,117 ms
コンパイル使用メモリ 277,496 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-06 00:33:51
合計ジャッジ時間 7,391 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 383 ms
6,820 KB
testcase_01 AC 359 ms
6,816 KB
testcase_02 AC 389 ms
6,816 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint1000000007;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000001
vector<mint> mul(vector<mint> a,vector<mint> b){
	
	vector<mint> c(a.size()+b.size()-1,0);
	rep(i,a.size()){
		rep(j,b.size()){
			c[i+j] += a[i]*b[j];
		}
	}
	return c;
}

mint Bostan_Mori(vector<mint> a,vector<mint> c,long long n){
	
	while(a.size()>=c.size())c.push_back(0);
	
	while(n!=0){
		auto cinv = c;
		for(int i=1;i<cinv.size();i+=2)cinv[i] *= -1;
		a = mul(a,cinv);
		c = mul(c,cinv);
		
		{
			vector<mint> na;
			for(int i=(n&1LL);i<a.size();i+=2)na.push_back(a[i]);
			swap(a,na);
		}
		{
			vector<mint> nc;
			for(int i=0;i<c.size();i+=2)nc.push_back(c[i]);
			swap(c,nc);
		}
		
		n>>=1;
	}
	return a[0];
}
int main(){
	
	long long n,p,c;
	cin>>n>>p>>c;
	vector<int> x= {2,3,5,7,11,13},y = {4,6,8,9,10,12};
	vector dp(p+1,vector(6,vector<mint>(13*p+1)));
	dp[0][0][0] = 1;
	rep(i,p+1){
		rep(j,6){
			rep(k,13*p+1){
				if(dp[i][j][k].val()==0)continue;
				if(j!=5)dp[i][j+1][k] += dp[i][j][k];
				if(i!=p)dp[i+1][j][k+x[j]] += dp[i][j][k];
			}
		}
	}
	{
		vector ndp(c+1,vector(6,vector<mint>(13*p+12*c+1)));
		rep(i,13*p+1){
			ndp[0][0][i] = dp[p][5][i];
		}
		dp = ndp;
	}
	rep(i,c+1){
		rep(j,6){
			rep(k,13*p+12*c+1){
				if(dp[i][j][k].val()==0)continue;
				if(j!=5)dp[i][j+1][k] += dp[i][j][k];
				if(i!=c)dp[i+1][j][k+y[j]] += dp[i][j][k];
			}
		}
	}
	vector<mint> C(13*p+12*c+1,0);
	rep(i,13*p+12*c+1)C[i] = dp[c][5][i];
	vector<mint> a(C.size(),0);
	for(int i=C.size()-1;i>=1;i--){
		a[i] = C[i];
		if(i!=C.size()-1)a[i] += a[i+1];
	}
	rep(i,C.size())C[i] *= -1;
	C[0] = 1;
	cout<<Bostan_Mori(a,C,n).val()<<endl;
	
	return 0;
}
0