結果
問題 | No.214 素数サイコロと合成数サイコロ (3-Medium) |
ユーザー | okaduki |
提出日時 | 2016-03-15 05:00:50 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 1,123 ms / 3,000 ms |
コード長 | 2,808 bytes |
コンパイル時間 | 1,440 ms |
コンパイル使用メモリ | 165,176 KB |
実行使用メモリ | 23,936 KB |
最終ジャッジ日時 | 2024-10-01 06:33:18 |
合計ジャッジ時間 | 5,743 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,123 ms
23,936 KB |
testcase_01 | AC | 729 ms
23,040 KB |
testcase_02 | AC | 660 ms
23,936 KB |
ソースコード
#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]; 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%2 == 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] += tmp[j]*p[j][i] % MOD; if(u[i] >= MOD) u[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] += p[0][j] * p[j][i] % MOD; if(next_e1[i] >= MOD) next_e1[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; LL dp1[7][51][MAX]; LL dp2[7][51][MAX]; int main(){ cin.tie(0); ios_base::sync_with_stdio(false); 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][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][j][n]) %= MOD; } int D = 13*P + 12*C+1; vector<LL> pat(D); REP(n1,MAX) REP(n2,MAX) (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; }