結果

問題 No.214 素数サイコロと合成数サイコロ (3-Medium)
ユーザー okadukiokaduki
提出日時 2016-03-15 05:23:50
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,581 ms / 3,000 ms
コード長 2,759 bytes
コンパイル時間 1,391 ms
コンパイル使用メモリ 164,592 KB
実行使用メモリ 22,272 KB
最終ジャッジ日時 2024-10-01 06:33:39
合計ジャッジ時間 6,737 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,581 ms
22,144 KB
testcase_01 AC 862 ms
21,632 KB
testcase_02 AC 904 ms
22,272 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<string> VS;
typedef pair<int, int> PII;
typedef long long LL;
typedef pair<LL, LL> PLL;

#define ALL(a)  (a).begin(),(a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define SZ(a) int((a).size())
#define EACH(i,c) for(typeof((c).begin()) i=(c).begin(); i!=(c).end(); ++i)
#define EXIST(s,e) ((s).find(e)!=(s).end())
#define SORT(c) sort((c).begin(),(c).end())

#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n)  FOR(i,0,n)

#define FF first
#define SS second
template<class S, class T>
istream& operator>>(istream& is, pair<S,T>& p){
  return is >> p.FF >> p.SS;
}

const double EPS = 1e-10;
const double PI  = acos(-1.0);
const LL MOD = 1e9+7;


typedef vector<LL> Col;
typedef vector<Col> Matrix;

LL A[1260][1260];
LL p[1260][1260];
LL u[1260];
LL next_e1[1260];
double t1,t2,t3,t4;
LL kitamasa(int K, LL n){
  u[0] = 1;
  REP(i,K-1) p[i][i+1] = 1;
  REP(i,K) p[K-1][i] = A[K-1][i];
  for(;n>0;n>>=1){
	if(n&1){
	  LL tmp[1260];
	  REP(i,K) tmp[i] = u[i];
	  for(int i=0;i<K;++i){
		u[i] = 0;
		for(int j=0;j<K;++j){
		  u[i] = (u[i]+tmp[j]*p[j][i]) % MOD;
		}
	  }
	}
	memset(next_e1, 0, sizeof(next_e1));
	for(int i=0;i<K;++i)
	  for(int j=0;j<K;++j){
		next_e1[i] = (next_e1[i] + p[0][j] * p[j][i]) % MOD;
	  }
	for(int i=0;i<K;++i)
	  p[0][i] = next_e1[i];
	for(int i=1;i<K;++i){
	  p[i][0] = p[i-1][K-1] * A[K-1][0] % MOD;
	  for(int j=1;j<K;++j)
		p[i][j] = (p[i-1][j-1] + p[i-1][K-1] * A[K-1][j]) % MOD;
	}
  }

  for(int i=0;i<K;++i)
	p[0][i] = u[i];
  for(int i=1;i<K;++i){
	p[i][0] = p[i-1][K-1] * A[K-1][0] % MOD;
	for(int j=1;j<K;++j){
	  p[i][j] = (p[i-1][j-1] + p[i-1][K-1] * A[K-1][j]) % MOD;
	}
  }

  LL res = 0;
  REP(i,K){
	res += p[K-1][i];
	if(res >= MOD) res -= MOD;
  }
  return res;
}

const int MAX = 660;
int dp1[7][51][MAX];
int dp2[7][51][MAX];
int main(){
  VI ps{2,3,5,7,11,13};
  VI qs{4,6,8,9,10,12};
  LL N, P, C; cin >> N >> P >> C;

  dp1[0][0][0] = 1;
  for(int i=0;i<6;++i)
	for(int j=0;j<=P;++j)
	  for(int k=j;k<=P;++k){
		int tmp = ps[i]*(k-j);
		for(int n=0;n+tmp<MAX;++n){
		  dp1[i+1][k][n+tmp] = (dp1[i+1][k][n+tmp]+dp1[i][j][n])%MOD;
		}
	  }
  
  dp2[0][0][0] = 1;
  for(int i=0;i<6;++i)
	for(int j=0;j<=C;++j)
	  for(int k=j;k<=C;++k){
		int tmp = qs[i]*(k-j);
		for(int n=0;n+tmp<MAX;++n){
		  dp2[i+1][k][n+tmp] = (dp2[i+1][k][n+tmp]+dp2[i][j][n])%MOD;
		}
	  }
  int D = 13*P + 12*C+1;
  vector<LL> pat(D);
  REP(n1,MAX) REP(n2,MAX)
	pat[n1+n2] = (pat[n1+n2] + dp1[6][P][n1] * dp2[6][C][n2]) % MOD;
  
  REP(i,D-1) A[i][i+1] = 1;
  FOR(i,1,D) A[D-1][D-i] = pat[i];

  cout << kitamasa(D, N) << endl;
  
  return 0;
}
0