結果

問題 No.213 素数サイコロと合成数サイコロ (3-Easy)
ユーザー motimoti
提出日時 2015-07-13 20:05:44
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 273 ms / 3,000 ms
コード長 1,612 bytes
コンパイル時間 540 ms
コンパイル使用メモリ 70,964 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-08 07:03:10
合計ジャッジ時間 1,358 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 104 ms
5,248 KB
testcase_01 AC 273 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <set>
#include <map>
#include <vector>
#include <array>

using namespace std;

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

typedef long long ll;

int const MOD = 1e9+7;

namespace Matrix
{
  typedef vector<ll> Vec;
  typedef vector<Vec> Mat;

  Mat operator * (Mat& A, Mat& B) {
    Mat C(A.size(), Vec(B[0].size()));
    rep(i, A.size()) rep(k, B.size()) rep(j, B[0].size())
      C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;
    return C;
  }

  void operator *= (Mat& A, Mat& B) {
    A = A*B;
  }

  Mat operator ^ (Mat& A, ll n) {
    Mat B(A.size(), Vec(A.size()));
    rep(i, A.size()) {
      B[i][i] = 1;
    }
    while(n > 0) {
      if(n & 1) { B *= A; }
      A *= A;
      n >>= 1;
    }
    return B;
  }
}

int dp_p[6][70];
int dp_q[6][70];
int dp_r[200];

int main() {

  constexpr array<int, 6> P = {2,3,5,7,11,13}, Q = {4,6,8,9,10,12};

  ll N;     cin >> N;
  int p, q; cin >> p >> q;

  dp_p[0][0] = 1;
  rep(i, P.size()) rep(time, p) {
    rep(sum, P.back()*time+1) dp_p[time+1][sum+P[i]] += dp_p[time][sum];
  }
  dp_q[0][0] = 1;
  rep(i, Q.size()) rep(time, q) {
    rep(sum, Q.back()*time+1) dp_q[time+1][sum+Q[i]] += dp_q[time][sum];
  }

  rep(i, P.back()*p+1) rep(j, Q.back()*q+1) {
    dp_r[i+j] += dp_p[p][i] * dp_q[q][j];
  }

  const int Max = p * 13 + q * 12;

  Matrix::Mat mat(Max, Matrix::Vec(Max));
  rep(i, Max-1) mat[i][i+1] = 1;
  REP(i, 1, Max+1) mat[Max-1][Max-i] = dp_r[i];

  mat = Matrix::operator^(mat,N+Max-1);

  ll ans = 0;
  rep(i, Max) (ans += mat[0][i]) %= MOD;
  cout << ans << endl;

  return 0;
}
0